Abe Leininger
|
3835a428c5
|
Adds nuclear norm support (#1894)
* adjust norm unit test tolerance
|
2025-03-04 13:26:02 -08:00 |
|
Awni Hannun
|
516ded618b
|
Dynamic slicing (#1741)
* dynamic slice and slice update
* python bindings + tests + fix set item
* fix compile issue
* comment
* fix jit
|
2025-01-07 14:02:16 -08:00 |
|
Awni Hannun
|
e03f0372b1
|
More shape type (#1705)
* more shape type
* fix
|
2024-12-19 08:08:20 -08:00 |
|
Awni Hannun
|
40c62c1321
|
Use int64 stride everywhere (#1671)
* use int64 stride everywhere
* fix ext
* fix ext
* more shape + cleanup
* one more
* few more
|
2024-12-09 11:09:02 -08:00 |
|
Alex Barron
|
048fabdabd
|
Fix vmap constant output size (#1524)
* use inputs to determine output size
* remove noop vmap tests
|
2024-10-30 16:16:53 -07:00 |
|
nicolov
|
8c9f0278b9
|
Add vmap to scatter (#1200)
* Add vmap to scatter
* updates
* vmap updates + a few more tests
* bug fix
---------
Co-authored-by: Awni Hannun <awni@apple.com>
|
2024-08-05 20:12:27 -07:00 |
|
nicolov
|
105d236889
|
Add vmap for SVD and inverse (#849)
|
2024-03-21 13:18:27 -07:00 |
|
Rifur13
|
126c9869c8
|
Implement the 'where' primitive for conditional selection (#664)
|
2024-02-22 15:10:48 -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 |
|
Awni Hannun
|
e5851e52b1
|
Add move and swap axis, and vmap for slice, concat, and gather (#158)
* add move and swap axis, and vmap for slice, concat, and gather
|
2023-12-14 12:59:12 -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 |
|