XXXXRT666 46d0fdc5ec Type Enhancement for Func Transforms and Bug Fix (#3003) ...

Co-authored-by: Awni Hannun <awni@apple.com>

2026-01-20 07:19:57 +09:00

..

_distributed_utils

Reverts changing the MLX_IBV_DEVICES to MLX_JACCL_DEVICES (#2999)

2026-01-14 15:44:17 -08:00

nn

Fix sharding of quantized models with non-power-of-2 bits (#3006)

2026-01-18 07:21:56 -08:00

optimizers

fix cross entropy axis param (#2641)

2025-10-01 16:49:55 -07:00

__main__.py

MLX in C++ example (#1736)

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

_reprlib_fix.py

override class function (#1418)

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

_stub_patterns.txt

Type Enhancement for Func Transforms and Bug Fix (#3003)

2026-01-20 07:19:57 +09: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

Added support for pytree types that inherit from tuple and typing.namedtuple (#2845)

2025-12-04 11:06:45 -08:00