Examples#
This template includes examples that use either Jax, PyTorch, or both!
| Example link | Research Area | Reference link | Frameworks |
|---|---|---|---|
| Image Classification | Supervised Learning (image classification) | ImageClassifier |
Torch + ⚡ |
| Image Classification (Jax) | Supervised Learning (image classification) | JaxImageClassifier |
Torch + Jax + ⚡ |
| Text Classification | NLP (text classification) | TextClassifier |
Torch + 🤗 + ⚡ |
| Reinforcement Learning (Jax) | RL | JaxRLExample |
Jax |
| LLM Fine-tuning | NLP (Causal language modeling) | LLMFineTuningExample |
Torch + 🤗 + ⚡ |