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