Merge pull request #5856 from google:autodidax

PiperOrigin-RevId: 361722343
This commit is contained in:
jax authors 2021-03-08 20:50:15 -08:00
commit 0b88b0ea9b
3 changed files with 3488 additions and 650 deletions

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff