Examples#
This template includes examples that use either Jax, PyTorch, or both!
| Example link | Research Area | Reference link | Frameworks |
|---|---|---|---|
| ExampleAlgorithm | Supervised Learning (image classification) | ExampleAlgorithm |
Torch + ⚡ |
| JaxExample | Supervised Learning (image classification) | JaxExample |
Torch + Jax + ⚡ |
| HFExample | NLP (text classification) | HFExample |
Torch + 🤗 + ⚡ |
| JaxRLExample | RL | JaxRLExample |
Jax |