zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-31 16:21:27 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
Files
199aebcf77d5342496947df2a7b7570245cc6218
mlx/python
History
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
..
mlx
Custom VJP and checkpointing (#541)
2024-01-30 16:04:45 -08:00
src
Custom VJP and checkpointing (#541)
2024-01-30 16:04:45 -08:00
tests
Make shape a tuple (#591)
2024-01-30 13:11:01 -08:00