mlx/python
Alex Barron b96e105244
Add grid_sample example to metal_kernel docs (#1352)
* Add `zero_outputs` and `atomic_outputs` options to `metal_kernel`

* add grid sample to docs

* zero_outputs -> init_value

* add missing header for linux
2024-08-23 18:24:16 -07:00
..
mlx Fix optimizer reloading from checkpoint (#1329) 2024-08-15 07:33:23 -07:00
src Add grid_sample example to metal_kernel docs (#1352) 2024-08-23 18:24:16 -07:00
tests Further reduction tuning (#1349) 2024-08-23 10:35:25 -07:00