Chaoran Yu b054838780 Added clarification to apply_fn parameter of apply_to_modules (#2831) ...

Co-authored-by: Awni Hannun <awni@apple.com>

2025-11-26 15:40:56 -08:00

..

__init__.py

Adding Relu2 (#2582)

2025-09-10 07:24:30 -07:00

activations.py

Adding Relu2 (#2582)

2025-09-10 07:24:30 -07:00

base.py

Added clarification to apply_fn parameter of apply_to_modules (#2831)

2025-11-26 15:40:56 -08: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

Add mode parameter for quantization (#2499)

2025-08-28 06:45:26 -07:00

linear.py

Add mode parameter for quantization (#2499)

2025-08-28 06:45:26 -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

Faster contiguous gather for indices in the first axis (#2552)

2025-08-28 21:26:30 -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