Back to List
Notice:This resource is provided by a third-party author. Please review the code with AI tools or manually before use to ensure security and compatibility.
Pythongoogle/paxml

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.

74.5/100
549Forks: 72
View on GitHub
Loading report...

Similar Projects

maxtext

85

A simple, performant and scalable Jax LLM!

Python2.3K

LightRAG

92

[EMNLP2025] "LightRAG: Simple and Fast Retrieval-Augmented Generation"

Python34.1K

VLMEvalKit

82

Open-source evaluation toolkit of large multi-modality models (LMMs), support 220+ LMMs, 80+ benchmarks

Python4.1K

verbalized-sampling

54

Verbalized Sampling, a training-free prompting strategy to mitigate mode collapse in LLMs by requesting responses with probabilities. Achieves 2-3x diversity improvement while maintaining quality. Model-agnostic framework with CLI/API for creative writing, synthetic data generation, and dialogue simulation.

Python739
Back to List