jax
bb16d5ac - fix bug with jax.remat static_argnums not supporting int

Commit

1 year ago

fix bug with jax.remat static_argnums not supporting int PiperOrigin-RevId: 707600082

References

Author

mattjj

Committer

Google-ML-Automation

Parents