flax
dcf120b6 - Removed private jax._src.api_util import and copied them into flax

Commit

45 days ago

Removed private jax._src.api_util import and copied them into flax

References

#4996 - Fix NNX jit static args with in_shardings issue #4989

Author

vfdev-5

vfdev-5

Committer

vfdev-5

vfdev-5

Parents

Loading