Awni Hannun 195b429d99 Put along axis + fixe for partition grad (#1430) ...

* put along axis, fixes for partition grad

* zeros for arg reduce

2024-09-23 10:03:38 -07:00

..

_static

Add Metal debug option and capture functions (#707)

2024-03-28 09:40:31 -07:00

_templates

Quantize embedding (#994)

2024-04-15 16:42:10 -07:00

cpp

Cpp docs (#1036)

2024-04-26 12:56:05 -07:00

dev

Simplifications for MLX C (#1396)

2024-09-06 19:16:50 -07:00

examples

Fix docs latex build and nits (#1297)

2024-07-29 11:44:06 -07:00

python

Put along axis + fixe for partition grad (#1430)

2024-09-23 10:03:38 -07:00

usage

Add docs for the distributed namespace (#1184)

2024-06-06 11:37:00 -07:00

conf.py

Fix docs latex build and nits (#1297)

2024-07-29 11:44:06 -07:00

index.rst

Custom Metal Kernels from Python (#1325)

2024-08-22 13:46:29 -07:00

install.rst

Xcode 160 (#1384)

2024-09-10 15:15:17 -07:00