Pinned Loading

1. google/paxml google/paxml Public

   Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…

   Python 463 69

2. trlx trlx Public

   Forked from CarperAI/trlx

   A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

   Python

3. google/praxis google/praxis Public

   Python 180 42

4. maxtext maxtext Public

   Forked from AI-Hypercomputer/maxtext

   A simple, performant and scalable Jax LLM!

   Python

5. saxml saxml Public

   Forked from google/saxml

   Python

6. accelerate accelerate Public

   Forked from huggingface/accelerate

   🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

   Python