Awni Hannun e7a2a3dcd1 Fix a couple bugs (#1161) ...

* fix jit reduce for RMS norm

* make strides a single buffer

* better eval error message

* fix compiling with inf and bf16

* fix cpu compile with bf16

2024-05-28 15:18:18 -07:00

..

mlx

Some fixes in docs (#1141)

2024-05-20 11:51:47 -07:00

src

Option to JIT steel gemm / conv (#1139)

2024-05-23 18:07:34 -07:00

tests

Fix a couple bugs (#1161)

2024-05-28 15:18:18 -07:00