mlx/python
jiyzhang 95e335db7b
Update smooth_l1_loss in losses.py (#1974)
According the definition of smooth_l1_loss, the line 

diff = predictions - targets

Should be updated to 

diff = mx.abs(predictions - targets)

After the modification, the result is consistent with PyTorch smooth_l1_loss
2025-03-19 20:19:02 -07:00
..
mlx Update smooth_l1_loss in losses.py (#1974) 2025-03-19 20:19:02 -07:00
src fix grad with inplace updates (#1961) 2025-03-13 19:13:09 -07:00
tests Use same accumulation precision in gemv as gemm (#1962) 2025-03-16 07:13:24 -07:00