mlx/python/mlx/nn/layers
2023-12-08 11:31:47 -08:00
..
__init__.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
activations.py NPY loading segfault bug (#34) 2023-12-06 12:03:47 -08:00
base.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -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 copyright + ack 2023-11-30 11:12:53 -08:00
embedding.py copyright + ack 2023-11-30 11:12:53 -08:00
linear.py copyright + ack 2023-11-30 11:12:53 -08:00
normalization.py copyright + ack 2023-11-30 11:12:53 -08:00
positional_encoding.py copyright + ack 2023-11-30 11:12:53 -08:00
transformer.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00