zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-10 03:06:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
eebd7c275d
mlx/mlx
History
Diogo dc2edc762c
added tri / tril / triu (#170) 
* added tri / tril / triu

* fixed tests

* ctest tests

* tri overload and simplified tests

* changes from comment

* more tests for m

* ensure assert if not 2-D

* remove broadcast_to

* minor tweaks

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-15 17:30:34 -08:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Lazy metal_device_ initialization (#185) 2023-12-15 16:06:46 -08:00
types copyright + ack 2023-11-30 11:12:53 -08:00
allocator.cpp copyright + ack 2023-11-30 11:12:53 -08:00
allocator.h copyright + ack 2023-11-30 11:12:53 -08:00
array.cpp copyright + ack 2023-11-30 11:12:53 -08:00
array.h add base kwarg to rope (#186) 2023-12-15 16:47:59 -08:00
CMakeLists.txt angelos's commit files 2023-11-29 10:42:59 -08:00
device.cpp copyright + ack 2023-11-30 11:12:53 -08:00
device.h copyright + ack 2023-11-30 11:12:53 -08:00
dtype.cpp copyright + ack 2023-11-30 11:12:53 -08:00
dtype.h random generation fix (#80) 2023-12-08 10:40:57 -08:00
fft.cpp copyright + ack 2023-11-30 11:12:53 -08:00
fft.h copyright + ack 2023-11-30 11:12:53 -08:00
graph_utils.cpp copyright + ack 2023-11-30 11:12:53 -08:00
graph_utils.h copyright + ack 2023-11-30 11:12:53 -08:00
load.cpp copyright + ack 2023-11-30 11:12:53 -08:00
load.h copyright + ack 2023-11-30 11:12:53 -08:00
mlx.h copyright + ack 2023-11-30 11:12:53 -08:00
ops.cpp added tri / tril / triu (#170) 2023-12-15 17:30:34 -08:00
ops.h added tri / tril / triu (#170) 2023-12-15 17:30:34 -08:00
primitives.cpp Add move and swap axis, and vmap for slice, concat, and gather (#158) 2023-12-14 12:59:12 -08:00
primitives.h accross -> across (#183) 2023-12-15 13:46:50 -08:00
random.cpp random generation fix (#80) 2023-12-08 10:40:57 -08:00
random.h copyright + ack 2023-11-30 11:12:53 -08:00
scheduler.cpp copyright + ack 2023-11-30 11:12:53 -08:00
scheduler.h copyright + ack 2023-11-30 11:12:53 -08:00
stream.h copyright + ack 2023-11-30 11:12:53 -08:00
transforms_impl.h format 2023-11-30 11:50:50 -08:00
transforms.cpp copyright + ack 2023-11-30 11:12:53 -08:00
transforms.h copyright + ack 2023-11-30 11:12:53 -08:00
utils.cpp Added mx.stack c++ frontend impl (#123) 2023-12-14 13:21:19 -08:00
utils.h Added mx.stack c++ frontend impl (#123) 2023-12-14 13:21:19 -08:00