zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-02 01:08:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
cb9e585b8ef4eac91299454def769a50e52e2483
mlx/python/tests
History
Kai Ma cb9e585b8e Style fix for loss functions (#91) 
* 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

* style
2023-12-08 21:11:56 -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
Style fix for loss functions (#91)
2023-12-08 21:11:56 -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