Pinned

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 399 60

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 144 39

4. maxtext maxtext Public

   Forked from google/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