jax
ec637fe4
- Merge pull request #29456 from j-towns:pjit-p-extend-primitives
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
232 days ago
Merge pull request #29456 from j-towns:pjit-p-extend-primitives PiperOrigin-RevId: 771132039
References
#29456 - Add `pjit_p` to `jax.extend.core.primitives`
Author
Google-ML-Automation
Parents
64c95744
7b7a5d8c
Loading
