
paxml
by google (organization)
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. (description from the project’s GitHub page)
Repository facts
- Stars
- 560
- Forks
- 77
- Open issues and pull requests
- 30
- Primary language
- Python
- License
- Apache License 2.0 (Apache-2.0)
- Owner
- google · Organization
- Created
- Jun 14, 2022
- Last push
- Oct 15, 2026
Topics
Related projects
Projects that share this one’s GitHub topics.
Rank 3,107: AI-Hypercomputer/maxtext
A simple, performant, and scalable Jax LLM!
2,445 starsPythonApache-2.0Last push Oct 18, 2026
Rank 64: hiyouga/LlamaFactory
Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)
75,561 starsPythonApache-2.0Last push Oct 10, 2026
Rank 157: HKUDS/LightRAG
[EMNLP2025] LightRAG: Simple and Fast Retrieval-Augmented Generation
40,199 starsPythonMITLast push Oct 15, 2026
Rank 1,801: open-compass/VLMEvalKit
Open-source evaluation toolkit of large multi-modality models (LMMs), support 220+ LMMs, 80+ benchmarks
4,445 starsPythonApache-2.0Last push Oct 16, 2026
Rank 2,905: om-ai-lab/OmAgent
[EMNLP-2024] Build multimodal language agents for fast prototype and production
2,669 starsPythonApache-2.0Last push Oct 16, 2026
Rank 3,069: 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.
2,482 starsPythonApache-2.0Last push Oct 10, 2024