zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-28 09:09:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
641d316484115bea9b58047fb9f15ce7749abd47
mlx/python/tests
History
Kai Ma 641d316484 MLE and L1 loss functions (#88) 
* MLE and L1 loss functions

* logsoftmax change and tests

* subtract max logit for numerical stability

* l1 name change

* cross entropy reduction + unit tests

* docstrings

* l1 test name change

* old loss impl + default none
2023-12-08 20:21:37 -08:00
..
mlx_tests.py
copyright + ack
2023-11-30 11:12:53 -08:00
test_array.py
Add isort pre-commit and run (#68)
2023-12-08 11:31:47 -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
MLE and L1 loss functions (#88)
2023-12-08 20:21:37 -08:00
test_ops.py
Add the remainder op (#85)
2023-12-08 15:08:52 -08:00
test_optimizers.py
Add isort pre-commit and run (#68)
2023-12-08 11:31:47 -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