Awni Hannun d03c01dfbc fix unflatten vjp (#1708)

2024-12-16 18:37:57 -08:00

..

mlx

No reshapes in quantized embedding (#1682)

2024-12-09 18:57:38 -08:00

src

Use expand_dims / unflatten / etc in more places (#1696)

2024-12-12 17:00:44 -08:00

tests

fix unflatten vjp (#1708)

2024-12-16 18:37:57 -08:00