mlx/python
Noah Kasmanoff de3d2467a3
Update: Fast GeLU Approximation (#744)
* add: fast gelu approx

* fix docs

* Update gelu_fast_approx function documentation

* Update python/mlx/nn/layers/activations.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* fix: test gelu

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
2024-02-26 21:08:50 -08:00
..
mlx Update: Fast GeLU Approximation (#744) 2024-02-26 21:08:50 -08:00
src Fix compile with non standard types (#745) 2024-02-26 19:28:53 -08:00
tests Update: Fast GeLU Approximation (#744) 2024-02-26 21:08:50 -08:00