Awni Hannun 111f1e71af Faster contiguous gather for indices in the first axis (#2552) ...

* faster contiguous gather for indices in the first axis

* work per thread > 1

* angelos suggestion for scales / biases

2025-08-28 21:26:30 -07:00

..

nn

Faster contiguous gather for indices in the first axis (#2552)

2025-08-28 21:26:30 -07:00

optimizers

docs: fix adam and adamw eps placement (#2416)

2025-07-24 16:40:45 -07:00

__main__.py

MLX in C++ example (#1736)

2025-01-02 19:09:04 -08:00

_os_warning.py

Version bump and os error (#807)

2024-03-07 13:57:58 -08:00

_reprlib_fix.py

override class function (#1418)

2024-09-16 13:21:04 -07:00

_stub_patterns.txt

fix submodule stubs (#1492)

2024-10-15 16:23:37 -07:00

distributed_run.py

nccl default for backend=any (#2528)

2025-08-22 12:24:27 -07:00

extension.py

Build with all cpu cores by default (#2336)

2025-07-07 06:06:45 -07:00

py.typed

Add py.typed to support PEP-561 (type-hinting) for mlx (#588)

2024-01-31 12:05:42 -08:00

utils.py

Support destination arg in tree flatten/unflatten (#2450)

2025-08-06 15:34:59 -07:00