mirror of
https://github.com/ml-explore/mlx.git
synced 2025-08-09 18:56:39 +08:00
4a64d4bff1
* Fix the weight shape for grouped convolutions from the nn API. * Add tests. * Pre-commit formatting. * Add input validation. * Use integer division instead of casting. * docs * nit --------- Co-authored-by: Awni Hannun <awni@apple.com> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| activations.py | ||
| base.py | ||
| containers.py | ||
| convolution_transpose.py | ||
| convolution.py | ||
| dropout.py | ||
| embedding.py | ||
| linear.py | ||
| normalization.py | ||
| pooling.py | ||
| positional_encoding.py | ||
| quantized.py | ||
| recurrent.py | ||
| transformer.py | ||
| upsample.py | ||