Awni Hannun cfb6a244ea allow parameters to be deleted (#2325)		2025-07-01 21:27:23 -07:00
..
__init__.py	Distributed layers (#1270)	2025-03-21 13:52:17 -07:00
activations.py	Fix module update in strict mode (#2321)	2025-06-29 11:12:29 -07:00
base.py	allow parameters to be deleted (#2325)	2025-07-01 21:27:23 -07:00
containers.py	copyright + ack	2023-11-30 11:12:53 -08:00
convolution_transpose.py	Added output_padding parameters in conv_transpose (#2092)	2025-04-23 09:26:33 -07:00
convolution.py	Ensure Conv2D and Conv3D's kernel sizes aren't trimmed (#1852)	2025-02-10 06:27:01 -08:00
distributed.py	Distributed layers (#1270)	2025-03-21 13:52:17 -07:00
dropout.py	Typing the dropout. (#1479)	2024-10-15 06:45:46 -07:00
embedding.py	Block sparse qmm (#1124)	2024-05-16 15:24:14 -07:00
linear.py	Block sparse qmm (#1124)	2024-05-16 15:24:14 -07:00
normalization.py	faster group norm (#1304)	2024-08-01 12:49:23 -07:00
pooling.py	Doc fix (#1615)	2024-11-22 11:12:25 -08:00
positional_encoding.py	Doc error for default for scale in SinusoidalPositionalEncoding (#1174)	2024-06-02 13:42:45 -07:00
quantized.py	Allow quant layer to be unfrozen (#2142)	2025-04-30 09:08:29 -07:00
recurrent.py	Faster RNN layers (#1419)	2024-09-17 06:04:19 -07:00
transformer.py	use sdpa and exportable functions in transformer multi head attention (#1760)	2025-01-09 13:11:55 -08:00
upsample.py	Nearest upsample (#2202)	2025-05-19 11:23:38 -07:00