Nikhil Mehta 0b7d71fd2f Add softmin, hardshrink, hardtanh (#1180) ...

---------

Co-authored-by: Nikhil Mehta <nikmehta@tesla.com>

2024-06-04 15:48:18 -07:00

..

nn

Add softmin, hardshrink, hardtanh (#1180)

2024-06-04 15:48:18 -07:00

optimizers

Treate 'minimum' differently in cosine decay (#1138)

2024-05-20 08:00:48 -07:00

_os_warning.py

Version bump and os error (#807)

2024-03-07 13:57:58 -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

feat: implement clip_grad_norm (#1043)

2024-05-03 09:07:02 -07:00