mlx/python
2024-01-31 12:55:30 -08:00
..
mlx Change the transformer to norm_first by default (#599) 2024-01-31 12:55:30 -08:00
src Custom VJP and checkpointing (#541) 2024-01-30 16:04:45 -08:00
tests Add GPU support for uint64/int64 reductions (#569) 2024-01-31 11:18:04 -08:00