jax
eff56af4
- [hijax] upgrade VJPHiPrimitive.def_bwd to work with accums
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
318 days ago
[hijax] upgrade VJPHiPrimitive.def_bwd to work with accums Co-authored-by: Dougal Maclaurin <dougalm@google.com>
References
#33041 - [hijax] upgrade VJPHiPrimitive.def_bwd to work with accums
Author
mattjj
Committer
mattjj
Parents
afbd5ff1
Loading
