jax
1ab4a2ea - Merge pull request #308 from google/pjit

Commit
7 years ago
Merge pull request #308 from google/pjit update pjit: full closure, nesting, grad of pjit
Author
Loading