Language Models
young-geng/EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
🧠 Natural Language Processing
PythonApache-2.0updated Aug 13, 2024
🏆 #5,463 overall#1134 of 1,375 in Language Models
young-geng /scalax
A simple library for scaling up JAX programs
$ git clone https://github.com/young-geng/scalax.gitStars
149
149
This week
—
Forks
11
11
Language
Python
License
Apache-2.0
Created
Jul 25, 2023
3.3 years old
Last push
Nov 4, 2025
Categories
GitHub topics
More in Language Models
Language Models
AI-Hypercomputer/maxtext🔥 active
▲ +3 ★ this week
A simple, performant, and scalable Jax LLM!
PythonApache-2.0updated Nov 13, 2026
Language Models
google/paxml
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 leading model flop utilization rates.
PythonApache-2.0updated Oct 26, 2026
Language Models
AI-Hypercomputer/JetStream
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
PythonApache-2.0updated Jan 5, 2026
Language Models
luyug/GradCache
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
PythonApache-2.0updated Mar 26, 2024
Language Models
BorealisAI/flora-opt
This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.
PythonLGPL-3.0updated Jul 1, 2024
Data from GitHub · snapshot Nov 13, 2026