fix: pin jax/jaxlib/flax lower bounds to avoid linear_util ImportError#39
Closed
ayaangazali wants to merge 1 commit into
Closed
fix: pin jax/jaxlib/flax lower bounds to avoid linear_util ImportError#39ayaangazali wants to merge 1 commit into
ayaangazali wants to merge 1 commit into