Awni Hannun 42afe27e12 std and expm1 (#973) ... * std and expm1 * actually add expm1 * fix linux * fix vjp * relax tol for linux test * Add it to the compilable primitives --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>		2024-04-08 14:26:01 -07:00
..
arange.h	copyright + ack	2023-11-30 11:12:53 -08:00
arg_reduce.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
binary_two.h	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
binary.cpp	add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)	2024-03-25 12:32:59 -07:00
binary.h	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
CMakeLists.txt	Fast RMS Norm (#862)	2024-03-21 07:20:54 -07:00
compiled_cpu.cpp	Fix cpu compile (#934)	2024-04-01 17:37:12 -07:00
compiled_nocpu.cpp	Ios compile (#784)	2024-03-04 20:02:26 -08:00
compiled_preamble.h	Refactor CPU compile preamble (#708)	2024-02-19 06:12:53 -08:00
compiled.cpp	Fix cpu compile (#934)	2024-04-01 17:37:12 -07:00
compiled.h	Fix cpu compile (#934)	2024-04-01 17:37:12 -07:00
conv.cpp	Convolution update (#651)	2024-02-28 20:11:16 -08:00
copy.cpp	Add missing && when forwarding args (#925)	2024-03-29 06:48:29 -07:00
copy.h	Add a SliceUpdate op and primitive (#850)	2024-03-20 10:39:25 -07:00
default_primitives.cpp	std and expm1 (#973)	2024-04-08 14:26:01 -07:00
erf.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
fft.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
indexing.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
inverse.cpp	Add vmap for SVD and inverse (#849)	2024-03-21 13:18:27 -07:00
lapack_helper.h	Add SVD primitive (#809)	2024-03-12 12:30:11 -07:00
load.cpp	Spelling (#342)	2024-01-01 21:08:17 -08:00
make_compiled_preamble.sh	Refactor CPU compile preamble (#708)	2024-02-19 06:12:53 -08:00
ops.h	std and expm1 (#973)	2024-04-08 14:26:01 -07:00
primitives.cpp	std and expm1 (#973)	2024-04-08 14:26:01 -07:00
qrf.cpp	fixes for building and running on iOS (#619)	2024-02-04 12:29:17 -08:00
quantized.cpp	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00
reduce.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
reduce.h	Implement vjps for some primitives in the fast namespace (#883)	2024-03-26 16:35:34 -07:00
scan.cpp	add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)	2024-03-25 12:32:59 -07:00
select.cpp	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
softmax.cpp	Option for precise softmax (#953)	2024-04-04 08:32:35 -07:00
sort.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
svd.cpp	Add vmap for SVD and inverse (#849)	2024-03-21 13:18:27 -07:00
ternary.h	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
threefry.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
threefry.h	copyright + ack	2023-11-30 11:12:53 -08:00
unary.h	Refactor CPU compile preamble (#708)	2024-02-19 06:12:53 -08:00
utils.h	Add missing && when forwarding args (#894)	2024-03-25 14:55:54 -07:00