mlx/mlx/backend/metal/jit
Awni Hannun de5f38fd48
Custom logsumexp (#2028)
* initial custom logsumexp

* more tests

* comments + fix
2025-03-31 07:36:55 -07:00
..
includes.h Custom logsumexp (#2028) 2025-03-31 07:36:55 -07:00
indexing.h Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00