Anchen 48f6ca8c3a Add theta cache for Rope and mask cache for ALiBi (#375)

2024-01-07 00:22:58 -08:00

..

__init__.py

Fix the sigmoid module (#371)

2024-01-04 13:16:36 -08:00

activations.py

Fix the sigmoid module (#371)

2024-01-04 13:16:36 -08:00

base.py

revert copy (#366)

2024-01-04 10:43:29 -08:00

containers.py

copyright + ack

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

convolution.py

Updated default argument for stride to 1 in Conv2d() in the docstring (#22)

2023-12-06 07:17:58 -08:00

dropout.py

feat: Add Dropout3d layer to nn.layers (#313)

2023-12-31 14:01:21 -08:00

embedding.py

copyright + ack

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

linear.py

Fix the implementation of the Bilinear layer (#347)

2024-01-02 16:46:18 -08:00

normalization.py

implemented InstanceNorm (#244)

2024-01-03 12:21:15 -08:00

positional_encoding.py

Add theta cache for Rope and mask cache for ALiBi (#375)

2024-01-07 00:22:58 -08:00

quantized.py

Support for quantized matmul with w and w^T (#349)

2024-01-03 14:22:36 -08:00

transformer.py

Transformer fix (#167)

2023-12-27 08:48:36 -08:00