Awni Hannun
|
2a980a76ce
|
Add stats and limit to common allocator and enable tests (#1988)
* add stats to common allocator and enable tests
* linux memory and default
* fix
|
2025-03-21 12:28:36 -07:00 |
|
Awni Hannun
|
4e1994e9d7
|
move memory APIs into top level mlx.core (#1982)
|
2025-03-21 07:25:12 -07:00 |
|
Awni Hannun
|
2770a10240
|
fix grad with inplace updates (#1961)
|
2025-03-13 19:13:09 -07:00 |
|
Awni Hannun
|
0a5215693e
|
Fix grad copies (#1854)
* fix grad with copies
* add test
* add test
|
2025-02-11 15:26:42 -08:00 |
|
Awni Hannun
|
2a45056ba8
|
Cycle leak break (#1856)
* detect and break leaks in custom function
* detect and break leaks in custom function
|
2025-02-11 14:45:02 -08:00 |
|
Awni Hannun
|
af1b725fda
|
Fix a couple of slicing bugs (#1827)
* fix a few bugs
* fix conv grad
* speedup test
* comment
|
2025-02-05 19:50:08 -08:00 |
|
Awni Hannun
|
b7c9f1d38f
|
scatter axis + gather axis primitives (#1813)
* scatter axis + gather axis primitives
* add transforms
* comment
|
2025-01-31 20:48:08 -08:00 |
|
Awni Hannun
|
0c259961ac
|
matmul jvps (#1772)
|
2025-01-17 10:36:26 -08:00 |
|
Awni Hannun
|
33421c1dd3
|
Limit grad recursion depth by not recursing through non-grad inputs (#1764)
* limit grad recursion depth
* add grad of module test
|
2025-01-14 14:33:18 -08:00 |
|
Awni Hannun
|
6fa0501387
|
Fix concatenate/slice_update vjp + reduce binary size (#1735)
* fix concatenate vjp + reduce binary size
* also cast in slice update
|
2025-01-02 16:36:33 -08:00 |
|
Awni Hannun
|
d03c01dfbc
|
fix unflatten vjp (#1708)
|
2024-12-16 18:37:57 -08:00 |
|
Awni Hannun
|
3f86399922
|
Real and Imag (#1490)
* real and imag
* fix
* fix
|
2024-10-15 16:23:15 -07:00 |
|
Awni Hannun
|
195b429d99
|
Put along axis + fixe for partition grad (#1430)
* put along axis, fixes for partition grad
* zeros for arg reduce
|
2024-09-23 10:03:38 -07:00 |
|
Angelos Katharopoulos
|
5c1fa64fb0
|
Custom transforms (#1246)
|
2024-07-10 18:00:01 -07:00 |
|
Awni Hannun
|
fd1c08137b
|
stable cumprod grad at 0 (#1167)
|
2024-05-31 12:28:42 -07:00 |
|
Awni Hannun
|
e6418781ab
|
Fix logsumexp edge case (#740)
* fix logsumexp
* fix inf constant
* also fix power grad
* fix ternary dispatch
|
2024-02-25 08:39:55 -08:00 |
|
Awni Hannun
|
cb6156d35d
|
Fix eval in trace bugs (#612)
* Fix eval in trace bugs
* comment nit
|
2024-02-02 09:57:12 -08:00 |
|
Awni Hannun
|
b207c2c86b
|
Power VJP fix for 0 (#505)
|
2024-01-20 01:17:40 -08:00 |
|
Angelos Katharopoulos
|
d8fabaa12b
|
Split multi output (#461)
* Multi-output split primitive
* Add the multi-output split to the ArrayIterator
* Add some grad tests for split
|
2024-01-16 13:33:55 -08:00 |
|
Tristan Bilot
|
f44c132f4a
|
Add scatter_min VJP (#462)
|
2024-01-16 00:37:40 -08:00 |
|
Tristan Bilot
|
6022d4129e
|
scatter_max vjp + bindings + tests (#431)
Co-authored-by: DjamelMesbah <djamel.mesbah@adservio.fr>
|
2024-01-14 14:12:15 -08:00 |
|
Awni Hannun
|
3b4f066dac
|
Correct types for vjp + tests (#418)
* correct types for vjp + tests
* fix build + comment
|
2024-01-10 13:32:37 -08:00 |
|
Angelos Katharopoulos
|
a611b0bc82
|
Removes the retain_graph flag (#385)
* Adds global tracing flag
* Removes retain_graph in favor of is_tracer
|
2024-01-07 15:16:51 -08:00 |
|
Zach Schillaci
|
5b9be57ac3
|
Add isort pre-commit and run (#68)
|
2023-12-08 11:31:47 -08:00 |
|
Awni Hannun
|
46a39e5b1f
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
Awni Hannun
|
8ca7f9e8e9
|
awni's commit files
|
2023-11-29 10:30:41 -08:00 |
|