Chunyang Wen
|
04edad8c59
|
Add doc string for path (#1937)
|
2025-03-06 14:20:09 -08:00 |
|
Angelos Katharopoulos
|
9680f72cca
|
Add a multi optimizer (#1916)
|
2025-03-04 13:16:35 -08:00 |
|
Saanidhya
|
4e22a1dffe
|
In continuation to PR1243 to solve issue #1240 (#1365)
* Solves issue #1240
* Correction
* Update python/mlx/utils.py
* Update python/mlx/utils.py
---------
Co-authored-by: Awni Hannun <awni@apple.com>
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
|
2024-08-28 11:40:41 -07:00 |
|
Awni Hannun
|
291cf40aca
|
Some fixes to typing (#1371)
* some fixes to typing
* fix module reference
* comment
|
2024-08-28 11:16:19 -07:00 |
|
toji
|
6768c6a54a
|
Adding missing type hints (#1243)
* added type hints for `run`, `tree_map` and `tree_map_with_path`
* fix lint
---------
Co-authored-by: Awni Hannun <awni@apple.com>
|
2024-07-23 07:29:38 -07:00 |
|
Nripesh Niketan
|
79c859e2e0
|
feat: implement clip_grad_norm (#1043)
* feat: implement `clip_grad_norm`
* pre-commit
* Add test for clip_grad_norm function in test_optimizers.py
* small fixes
* fix
* lint
* Update tree_reduce
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Refactor clip_grad_norm function to include documentation and improve readability
* format docstring
* Add acknowlegements
* text wrap
* pre-commit
* nits in docs
---------
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
|
2024-05-03 09:07:02 -07:00 |
|
Awni Hannun
|
cd9e184529
|
Quantize embedding (#994)
* quantize embedding
* rename as_linear + comment
* consistency in docs
* fix test
|
2024-04-15 16:42:10 -07:00 |
|
Diogo
|
35431a4ac8
|
Adds device context manager (#679)
|
2024-02-14 14:14:58 -08:00 |
|
Chunyang Wen
|
2aedf3e791
|
Minor refactor for tree_map and tree_unflatten (#311)
* Minor refact for tree_map and tree_unflatten
* Remove the if statement
---------
Co-authored-by: Chunyang Wen <chunyang_wen@apple.com>
|
2023-12-28 20:55:10 -08:00 |
|
Chunyang Wen
|
473b6b43b4
|
Use defaultdict (#307)
Co-authored-by: Chunyang Wen <chunyang_wen@apple.com>
|
2023-12-28 14:46:13 -08:00 |
|
Angelos Katharopoulos
|
57fe918cf8
|
Adds C++ and nn quantization utilities (#230)
* Add C++ de-/quantize ops
* Add quantize functions to the docs and tests
* Add a QuantizedLinear module
|
2023-12-20 14:17:38 -08:00 |
|
Awni Hannun
|
db487e6b1a
|
format
|
2023-11-30 11:50:50 -08:00 |
|
Awni Hannun
|
46a39e5b1f
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
Angelos Katharopoulos
|
d1f86272a2
|
angelos's commit files
|
2023-11-29 10:42:59 -08:00 |
|