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.

75.4/100
550Forks: 70
View on GitHub
Loading report...

Similar Projects

maxtext

85

A simple, performant and scalable Jax LLM!

Python2.2K

LightRAG

92

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

Python29.1K

VLMEvalKit

81

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

Python3.9K

verbalized-sampling

59

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.

Python705
Back to List