Luca Vivona ab51eb80ad feat: support dict and list destinations in tree_flatten, add tree_unflatten

2025-07-29 17:25:57 -04:00

..

nn

allow parameters to be deleted (#2325)

2025-07-01 21:27:23 -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

Fix the launcher when ran locally (#2147)

2025-05-01 12:56:09 -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

feat: support dict and list destinations in tree_flatten, add tree_unflatten

2025-07-29 17:25:57 -04:00