Domů Procházet Nápověda
Přihlásit se
SunnyMirror
/
DeepLearningExamples
zrcadlo https://github.com/NVIDIA/DeepLearningExamples.git
2
0
Rozštěpit 0
Soubory Úkoly 0 Wiki
Strom: 41f582bd9f
Větve Značky
DeepLearningExa...
/
JAX
/
LanguageModeling
/
PAXML
Sharath Turuvekere Sreenivas 41f582bd9f [JAX] Add JAX models with reference to Rosetta Github před 2 roky
..
README.md 41f582bd9f [JAX] Add JAX models with reference to Rosetta Github před 2 roky

README.md

Paxml (aka Pax) is a framework for training LLMs. It allows for advanced and configurable experimentation and parallelization. It is based on JAX and Praxis.

PAXML on GPUs

Please refer to Rosetta PAXML, NVIDIA's project that enables seamless training of LLMs, CV models and multimodal models in JAX, for information about running models and experiments on GPUs in PAXML.