mlx/mlx
Bahaa ff2b58e299
Add support for repeat (#278)
* add repeat function

* fix styling

* optimizing repeat

* fixed minor issues

* not sure why that folder is there xD

* fixed now for sure

* test repeat not repeat test

* Fixed

---------

Co-authored-by: Bahaa Eddin tabbakha <bahaa@Bahaas-MacBook-Pro.local>
2023-12-27 13:11:38 -08:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Safetensor support (#215) 2023-12-27 02:06:55 -08:00
io Safetensor support (#215) 2023-12-27 02:06:55 -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 Safetensor support (#215) 2023-12-27 02:06:55 -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
linalg.cpp linalg.norm (#187) 2023-12-26 19:42:04 -08:00
linalg.h linalg.norm (#187) 2023-12-26 19:42:04 -08:00
mlx.h linalg.norm (#187) 2023-12-26 19:42:04 -08:00
ops.cpp Add support for repeat (#278) 2023-12-27 13:11:38 -08:00
ops.h Add support for repeat (#278) 2023-12-27 13:11:38 -08:00
primitives.cpp Improve names of quantization arguments (#235) 2023-12-20 16:53:53 -08:00
primitives.h Safetensor support (#215) 2023-12-27 02:06:55 -08:00
random.cpp random.uniform must respect dtype, even if lower precision than "low" (#280) 2023-12-24 07:04:43 -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 Revisit autorelease memory pools (#260) 2023-12-22 11:01:26 -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 implemented Flatten Module (#149) 2023-12-16 21:54:37 -08:00
utils.h Added mx.stack c++ frontend impl (#123) 2023-12-14 13:21:19 -08:00