Awni Hannun e88e474fd1 Reduce vmap + some fixes (#601)

2024-02-01 11:30:28 -08:00

..

mlx

Fix for AdaDelta (#603)

2024-02-01 09:56:27 -08:00

src

Custom VJP and checkpointing (#541)

2024-01-30 16:04:45 -08:00

tests

Reduce vmap + some fixes (#601)

2024-02-01 11:30:28 -08:00