mlx/python/mlx/nn
Nripesh Niketan 5ad8fb7268
feat: add softsign, softmax, hardswish, logsoftmax activation function (#309)
* feat: add softsign activation function

* run pre-commit

* Add Softsign activation function

* Add Softsign activation function

* Add documentation for ReLU6, Softplus, and Softsign activations

* Update activation functions in neural network layers

* Add LogSoftmax and Hardswish activations

* run pre-commit

* Update activations.py

* Added acknowledgements

* Fix activation function comments

* Fix activation functions in neural network layers
2023-12-29 11:49:36 -08:00
..
layers feat: add softsign, softmax, hardswish, logsoftmax activation function (#309) 2023-12-29 11:49:36 -08:00
__init__.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
losses.py implement-batch-norm-layer (#217) 2023-12-25 07:32:53 -08:00
utils.py copyright + ack 2023-11-30 11:12:53 -08:00