Jason b0cd092b7f Added activation functions: leaky_relu relu6 softplus elu celu logsigmoid (#108) ... * added leaky_relu relu6 softplus elu celu logsigmoid * minor fixes for docstring and benchmark imports * fixed elu implementation and added tests * added tests for optional param, changed leaky_relu param to fit pytorch documentation		2023-12-10 16:31:38 -08:00
..
mlx_tests.py	copyright + ack	2023-11-30 11:12:53 -08:00
test_array.py	Hashable dtype and mlx.core prefixed repr (#89)	2023-12-09 09:35:28 -08:00
test_autograd.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_bf16.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_blas.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_conv.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_device.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_eval.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_fft.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_load.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_nn.py	Added activation functions: leaky_relu relu6 softplus elu celu logsigmoid (#108)	2023-12-10 16:31:38 -08:00
test_ops.py	Bug fix in metal binary kernel dispatch for large arrays (#125)	2023-12-10 16:12:31 -08:00
test_optimizers.py	Pr template (#99)	2023-12-09 09:36:56 -08:00
test_random.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_reduce.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_tree.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_vmap.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00