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 + ⚡ |
| TextClassificationExample | NLP (text classification) | TextClassificationExample |
Torch + 🤗 + ⚡ |
| JaxRLExample | RL | JaxRLExample |
Jax |
| LLMFinetuningExample | NLP (Causal language modeling) | LLMFineTuningExample |
Torch + 🤗 + ⚡ |