Awni Hannun 657f466402 use sdpa and exportable functions in transformer multi head attention (#1760)

2025-01-09 13:11:55 -08:00

..

nn

use sdpa and exportable functions in transformer multi head attention (#1760)

2025-01-09 13:11:55 -08:00

optimizers

Support bias correction in Adam and AdamW optimizers (#1640)

2024-12-06 12:13:34 -08: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

extension.py

Adding missing type hints (#1243)

2024-07-23 07:29:38 -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

In continuation to PR1243 to solve issue #1240 (#1365)

2024-08-28 11:40:41 -07:00