jax
5566c346
- Merge pull request #32478 from jax-ml:newstyle-hiprimitive
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
125 days ago
Merge pull request #32478 from jax-ml:newstyle-hiprimitive PiperOrigin-RevId: 817311993
References
#32478 - Add an API for defining a new hijax primitive in terms of expand/fwd/bwd/batch
Author
Google-ML-Automation
Parents
227d7e24
4dcfaf53
Loading