mlx/python
Paul Paczuski ebd7135b50
Improve stability of BCE loss calculation for input probabilities close to or exactly 0 or 1 (#1280)
* Improve stability of BCE loss calculation

* Standardize comment

* Apply formatting with black via pre-commit

* Add usage recommendation to docstring

* Update python/mlx/nn/losses.py

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
2024-07-24 08:38:22 -07:00
..
mlx Improve stability of BCE loss calculation for input probabilities close to or exactly 0 or 1 (#1280) 2024-07-24 08:38:22 -07:00
src Implement sampling from laplace distribution. (#1279) 2024-07-24 15:15:37 +02:00
tests Improve stability of BCE loss calculation for input probabilities close to or exactly 0 or 1 (#1280) 2024-07-24 08:38:22 -07:00