AtomicVar 755dcf6137 Enable cross_entropy loss to handle dense targets (#517) ...

* Enable cross_entropy loss to handle dense targets

Dense targets means probabilities or one-hot encodings.

* better shape check of weights

* nits in docstring

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2024-01-23 12:17:22 -08:00

..

nn

Enable cross_entropy loss to handle dense targets (#517)

2024-01-23 12:17:22 -08:00

_reprlib_fix.py

copyright + ack

2023-11-30 11:12:53 -08:00

extension.py

Add isort pre-commit and run (#68)

2023-12-08 11:31:47 -08:00

optimizers.py

format (#510)

2024-01-20 10:33:46 -08:00

utils.py

Minor refactor for tree_map and tree_unflatten (#311)

2023-12-28 20:55:10 -08:00