zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-26 10:41:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
641d316484
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