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 |