mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 17:31:16 +08:00
295ce9db09
* Added an identity and bilinear layers Added a reset_parameters option Added normal init for bias * pre-commit run * add type hints for parameters and the return type change Bilinear math to x_1 and x_2 change __call__ arguments to x and y instead of input and output add explanation to the Initialization * Remove unnecessary reshape * Added 'i' to bilinear formula * Changed bilinear computation to two matrix multiplications * avoid saving intermediate results, kept y in bilinear for better clarity (can be replaced with x1) * Changed math formula in Linear Added more explanation to math formulas Changed x1, x2 reshape to support all inputs sizes |
||
|---|---|---|
| .. | ||
| mlx_tests.py | ||
| test_array.py | ||
| test_autograd.py | ||
| test_bf16.py | ||
| test_blas.py | ||
| test_conv.py | ||
| test_device.py | ||
| test_eval.py | ||
| test_fft.py | ||
| test_linalg.py | ||
| test_load.py | ||
| test_nn.py | ||
| test_ops.py | ||
| test_optimizers.py | ||
| test_quantized.py | ||
| test_random.py | ||
| test_reduce.py | ||
| test_tree.py | ||
| test_vmap.py | ||