Jax Loss

A jax library of common machine learning loss functions

Installation

pip install jax-loss

Loss Functions Included

• Binary Cross Entropy With Logits Loss

If you don't see the loss function you need, feel free to open an issue or open a PR!

Loss Functions to be Implemented

• L1
• Mean Squared Error
• Cross Entropy
• Negative Log Liklihood
• KL Divergance
• Binary Cross Entropy
• Triplet Margin
• Cosine Embedding Loss
• Hinge