2 Commits

Author	SHA1	Message	Date
Nripesh Niketan	ffff671273	Update pre-commit hooks (#984)	2024-04-11 07:27:53 -07:00
Angelos Katharopoulos	0de5988f92	Custom VJP and checkpointing (#541) ... * Implement custom_vjp and checkpointing * Add a dependency management primitive * Change the eval order to deep branches first * Add graph depth tracking to the array	2024-01-30 16:04:45 -08:00