Angelos Katharopoulos ba8d6bf365 Change the transformer to norm_first by default (#599)

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