Angelos Katharopoulos 0de5988f92 Custom VJP and checkpointing (#541) ... * Implement custom_vjp and checkpointing * Add a dependency management primitive * Change the eval order to deep branches first * Add graph depth tracking to the array		2024-01-30 16:04:45 -08:00
..
3rdparty	jagrit's commit files	2023-11-29 10:52:08 -08:00
backend	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
io	GGUF: Avoid dequantization when format is compatible (#426)	2024-01-23 15:43:57 -08:00
types	Spelling (#342)	2024-01-01 21:08:17 -08:00
allocator.cpp	Memory allocation (#292)	2024-01-02 11:59:19 -08:00
allocator.h	Memory allocation (#292)	2024-01-02 11:59:19 -08:00
array.cpp	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
array.h	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
CMakeLists.txt	Compile front-end (#476)	2024-01-26 13:45:30 -08:00
compile.cpp	Compile front-end (#476)	2024-01-26 13:45:30 -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	Spelling (#342)	2024-01-01 21:08:17 -08:00
fft.h	copyright + ack	2023-11-30 11:12:53 -08:00
graph_utils.cpp	Fix stub generation, change graph exporting for arrows to go to outputs (#455)	2024-01-14 14:06:16 -08:00
graph_utils.h	copyright + ack	2023-11-30 11:12:53 -08:00
io.h	GGUF: Load and save metadata (#446)	2024-01-19 14:06:05 -08:00
linalg.cpp	QR factorization (#310)	2024-01-26 09:27:31 -08:00
linalg.h	QR factorization (#310)	2024-01-26 09:27:31 -08:00
mlx.h	GGUF: Load and save metadata (#446)	2024-01-19 14:06:05 -08:00
ops.cpp	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
ops.h	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
primitives.cpp	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
primitives.h	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
random.cpp	Spelling (#342)	2024-01-01 21:08:17 -08:00
random.h	feat: add time based seed to random.h (#457)	2024-01-16 07:32:28 -08:00
scheduler.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
scheduler.h	move all ObjC (via metal-cpp) interaction until post static initializers (#370)	2024-01-04 16:12:00 -08:00
stream.h	copyright + ack	2023-11-30 11:12:53 -08:00
transforms_impl.h	Compile front-end (#476)	2024-01-26 13:45:30 -08:00
transforms.cpp	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
transforms.h	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
utils.cpp	Added formatter structure and a boolean value formatter (#354)	2024-01-18 07:49:41 -08:00
utils.h	Added formatter structure and a boolean value formatter (#354)	2024-01-18 07:49:41 -08:00