Awni Hannun fe1dabf272 Fix compile with non standard types (#745) ...

* refactor tree utils

* fix compile + tree code refactor

* Add an extra test

* add a few missing activations to docs

* hash structure

* Encode the full argument structure

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>

2024-02-26 19:28:53 -08:00

..

nn

Fix compile with non standard types (#745)

2024-02-26 19:28:53 -08:00

optimizers

Add linear warmup and schedule joining for use with existing schedules (#721)

2024-02-26 07:28:48 -08:00

_reprlib_fix.py

copyright + ack

2023-11-30 11:12:53 -08:00

extension.py

Add isort pre-commit and run (#68)

2023-12-08 11:31:47 -08: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

Adds device context manager (#679)

2024-02-14 14:14:58 -08:00