fix hparams self tuning jax

Niccolo-Ajroldi · web-flow · commit 4aba50946e5e · 2025-03-14T10:10:01.000+01:00
diff --git a/prize_qualification_baselines/self_tuning/jax_nadamw_full_budget.py b/prize_qualification_baselines/self_tuning/jax_nadamw_full_budget.py
@@ -1,6 +1,5 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in Jax."""
 
-import collections
 import functools
 
 # isort: off
