mlx/python
Cheng 1ba18ff7d9
[CUDA] Fix conv grads with groups (#2495)
* Put reshape utils in one file

* [CUDA] Fix conv grads with groups

* Put the reshape utils in gpu/copy.h
2025-08-16 10:09:18 +09:00
..
mlx Support destination arg in tree flatten/unflatten (#2450) 2025-08-06 15:34:59 -07:00
scripts fix cuda pypi package (#2423) 2025-07-25 15:20:29 -07:00
src Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
tests [CUDA] Fix conv grads with groups (#2495) 2025-08-16 10:09:18 +09:00