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