jax
51560bf3 - [JAX] [XLA:Python] Migrate pytree module to JAX.

Commit

1 year ago

[JAX] [XLA:Python] Migrate pytree module to JAX. PiperOrigin-RevId: 740142231

References

ci-upstream-sync-156_1

#27386 - [JAX] [XLA:Python] Migrate pytree module to JAX.

Author

hawkinsp

hawkinsp

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading