jax
da620a8e - working on nested pjit, misc bug fixes

Commit

7 years ago

working on nested pjit, misc bug fixes

References

#308 - update pjit: full closure, nesting, grad of pjit

Author

mattjj

mattjj

Committer

mattjj

mattjj

Parents

Loading