ruff linting issues

Author: init-22

algoperf/workloads/cifar/cifar_jax/input_pipeline.py

@@ -11,7 +11,6 @@
 import jax
 import tensorflow as tf
 import tensorflow_datasets as tfds
-from flax import jax_utils
 
 from algoperf import spec
 from algoperf.data_utils import shard_and_maybe_pad_np

algorithms/baselines/self_tuning/jax_nadamw_full_budget.py

@@ -16,7 +16,6 @@
 import jax
 import jax.numpy as jnp
 import optax
-from flax import jax_utils
 
 from algoperf import jax_sharding_utils, spec