autodidax: jit, multi-output, pytrees, DeviceArrays

This commit is contained in:
Matthew Johnson 2021-02-25 17:58:03 -08:00
parent ed16ad8ca2
commit 6b5133d71f
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