mlx/python
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