Skip to content

Commit

Permalink
Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into pax-te-docu…
Browse files Browse the repository at this point in the history
…mentation
  • Loading branch information
ashors1 committed Aug 29, 2023
2 parents d6809eb + 580f871 commit 913e7f9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -157,7 +157,7 @@ This repo currently hosts a public CI for JAX on NVIDIA GPUs and covers some JAX
We currently enable training and evaluation for the following models:
| Model Name | Pretraining | Fine-tuning | Evaluation |
| :--- | :---: | :---: | :---: |
| [GPT-3(paxml)](./rosetta/rosetta/projects/pax) | ✔️ | ✔️ | ✔️ |
| [GPT-3(paxml)](./rosetta/rosetta/projects/pax) | ✔️ | | ✔️ |
| [t5(t5x)](./rosetta/rosetta/projects/t5x) | ✔️ | ✔️ | ✔️ |
| [ViT](./rosetta/rosetta/projects/vit) | ✔️ | ✔️ | ✔️ |

Expand Down

0 comments on commit 913e7f9

Please sign in to comment.