mlx/python
Awni Hannun 4f46e9c997
More fixes for arrays with large sizes (#1405)
* compile works for big arrays when contiguous

* style

* nits in docs

* a bunch more stuff

* update jit

* update jit

* use constant for shapes and strides and remove elem_to_loc overload

* use kernel instantiation

* docs nits

* update binary and ternary

* comments
2024-09-17 12:46:31 -07:00
..
mlx Faster RNN layers (#1419) 2024-09-17 06:04:19 -07:00
src More fixes for arrays with large sizes (#1405) 2024-09-17 12:46:31 -07:00
tests Data parallel helper (#1407) 2024-09-16 18:17:21 -07:00