mlx/python
Cheng 8347575ba1
[CUDA] Implement Scan kernel (#2347)
* Contiguous scan

* Strided scan

* Enable tests

* Fix failing logaddexp test

* Use cexpf in Metal
2025-07-10 16:54:12 -07:00
..
mlx Fix type promotion in Adam with bias correction (#2350) 2025-07-10 11:14:42 -07:00
scripts Build CUDA release in Circle (#2306) 2025-06-19 15:26:36 -07:00
src MoE backward improvements (#2335) 2025-07-07 17:59:53 -07:00
tests [CUDA] Implement Scan kernel (#2347) 2025-07-10 16:54:12 -07:00