Awni Hannun 3dcb286baf Remove stream from average grads so it uses default (#2532) ... * Remove stream from average grads so it uses default * comment		2025-08-25 15:56:29 -07:00
..
nn	Remove stream from average grads so it uses default (#2532)	2025-08-25 15:56:29 -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