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