Cheng 5be5daa6ef Use compiled function in Sigmoid module (#1116)

2024-05-14 06:25:57 -07:00

..

__init__.py

Conv3d (#993)

2024-05-11 06:15:02 -07:00

activations.py

Use compiled function in Sigmoid module (#1116)

2024-05-14 06:25:57 -07:00

base.py

Support Chaining for some of functionalities of nn.Module (#885) (#897)

2024-03-27 19:58:29 -07:00

containers.py

copyright + ack

2023-11-30 11:12:53 -08:00

convolution.py

Conv3d (#993)

2024-05-11 06:15:02 -07:00

dropout.py

Make shape a tuple (#591)

2024-01-30 13:11:01 -08:00

embedding.py

Quantize embedding (#994)

2024-04-15 16:42:10 -07:00

linear.py

Reduce a little overhead (#871)

2024-03-22 17:29:36 -07:00

normalization.py

Reduce a little overhead (#871)

2024-03-22 17:29:36 -07:00

pooling.py

avgpool, not maxpool (#1002)

2024-04-17 08:26:22 -07:00

positional_encoding.py

No reshape rope (#838)

2024-03-18 17:03:07 -07:00

quantized.py

Use correct module type in quantized.py (#1115)

2024-05-14 06:25:42 -07:00

recurrent.py

fix a couple bugs (#952)

2024-04-02 12:07:41 -07:00

transformer.py

Change the transformer to norm_first by default (#599)

2024-01-31 12:55:30 -08:00

upsample.py

Upsample with bicubic interpolation (#967)

2024-04-10 15:47:22 -07:00