mlx/python/mlx/nn/layers
2024-06-02 13:42:45 -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 Some fixes in docs (#1141) 2024-05-20 11:51:47 -07:00
dropout.py Make shape a tuple (#591) 2024-01-30 13:11:01 -08: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 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 Doc error for default for scale in SinusoidalPositionalEncoding (#1174) 2024-06-02 13:42:45 -07:00
quantized.py Block sparse qmm (#1124) 2024-05-16 15:24:14 -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