David Koski 601c6d6aa8 Fix for AdaDelta (#603) ...

- state was being read from parameter "s"
- but being stored in parameter "u"

2024-02-01 09:56:27 -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

Add GPU support for uint64/int64 reductions (#569)

2024-01-31 11:18:04 -08:00