mlx/benchmarks/python
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
..
blas Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
comparative Added activation functions: leaky_relu relu6 softplus elu celu logsigmoid (#108) 2023-12-10 16:31:38 -08:00
batch_matmul_bench.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
llama_jax_bench.py copyright + ack 2023-11-30 11:12:53 -08:00
llama_mlx_bench.py copyright + ack 2023-11-30 11:12:53 -08:00
llama_torch_bench.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
single_ops.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
time_utils.py copyright + ack 2023-11-30 11:12:53 -08:00