Home/LLM Frameworks/maxtext

maxtext

AI-Hypercomputer/maxtext

A simple, performant, and scalable Jax LLM

GitHub Website

GraphCanon updated 2w · GitHub synced 2w

2.4k stars581 forksLast push 2w Python Apache-2.0

Decision brief

MaxText is a performant large language model built on the JAX framework, focusing on fine-tuning and scaling options for various architectures like GPT series, LLaMA family, Mistral, Mixtral.

Good fit when

Use MaxText when you require high-performance training and tuning over different deep learning architectures within a unified framework like JAX
Ideal for projects needing scalable solutions across multiple models including Gemma2, GPT series, LLaMA family, Mistral, Mixtral; leveraging JAX's parallel computation capabilities

Avoid when

Avoid using MaxText if you are only interested in TensorFlow or PyTorch specific optimizations and functionalities without a seamless transition to JAX
Not recommended for users requiring model customization outside of supported architectures as it strictly adheres to Gemma2, GPT, LLaMA series, Mistral, Mixtral

Hosting:: unknown - N/A as details on hosting are not provided in the repository

Observed Jul 15, 2026 · Source: enrich:decision_facts

maxtext alternatives Trust report

Verify the decision

Maintenance and security

Full trust report

Maintenance: Very active (0d since push); As of 2w
Provenance: Not a fork · Organization account; As of 2w
Security (OSV): No lockfile; As of 1mo

Public GitHub metadata and optional OSV scans. Signals, not a guarantee. Trust methodology.

Install

pip install maxtext

PyPI

Similar tools

Same-category neighbours. No typed graph edges are catalogued for this tool yet.

llmfit

Hundreds of models & providers. One command to find what runs on your hardware.

32k★

litgpt

High-performance LLMs with recipes for pretraining, finetuning and deployment

14k★

open-llms

A list of open LLMs available for commercial use.

13k★

Evidence and technical details

Sourced facts, taxonomy, compatibility claims, README excerpt, and machine-readable endpoints.

Overview

MaxText is an efficient large language model built on the JAX framework for deep learning. It supports multiple architectures including Gemma2, GPT, LLaMA series, Mistral, and Mixtral. Key areas of focus are fine-tuning and scaling options tailored for performance.

Capability facts

CLI: CLI entrypoint
Source: pyproject.toml:[project.scripts] · Aug 7, 2026
Languages: python
Source: github.language+pyproject.toml · Aug 7, 2026

README

Installation

See our installation guide to install MaxText with pip from PyPI.

License

Apache License 2.0

For agents

This page has a .md twin and JSON over the API.

/tools/ai-hypercomputer-maxtext.md GET /api/graphcanon/tools/ai-hypercomputer-maxtext GET /api/graphcanon/graph?tool=ai-hypercomputer-maxtext