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
🏆 #2,267 overall#517 of 1,375 in 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.
$ git clone https://github.com/google/paxml.gitStars
560
560
This week
—
Forks
73
73
Language
Python
License
Apache-2.0
Created
Jun 14, 2022
4.4 years old
Last push
Oct 26, 2026
Categories
More in Language Models
Language Models
HKUDS/LightRAG🔥 active
▲ +112 ★ this week
[EMNLP2025] LightRAG: Simple and Fast Retrieval-Augmented Generation
🔍 Search & Retrieval
PythonMITupdated Nov 13, 2026
Language Models
Helicone/helicone🔥 active
▲ +9 ★ this week
🧊 Open source LLM observability platform. One line of code to monitor, evaluate, and experiment. YC W23 🍓
TypeScriptApache-2.0updated Nov 12, 2026
Language Models
darrenburns/elia
A snappy, keyboard-centric terminal user interface for interacting with large language models. Chat with ChatGPT, Claude, Llama 3, Phi 3, Mistral, Gemma and more.
PythonApache-2.0updated Oct 10, 2024
Language Models
AI-Hypercomputer/maxtext🔥 active
▲ +3 ★ this week
A simple, performant, and scalable Jax LLM!
PythonApache-2.0updated Nov 13, 2026
Language Models
PKU-Alignment/safe-rlhf
▲ +3 ★ this week
Safe RLHF: Constrained Value Alignment via Safe Reinforcement Learning from Human Feedback
PythonApache-2.0updated Nov 24, 2025
Data from GitHub · snapshot Nov 13, 2026