AIAny
AI Infra2024
Icon for item

Grok-1

Open-weights 314B-parameter Mixture-of-Experts language model (8 experts, 2 active per token, 8,192-token context) released under Apache 2.0. Ships a raw JAX checkpoint plus reference inference code; needs heavy multi-GPU memory to load.

Introduction

Releasing 314 billion parameters under Apache 2.0 was, at the time, the largest open-weight model anyone had shipped — and the point was less the benchmarks than the precedent. What you get here is a raw base checkpoint, not the chat-tuned Grok that powers the product: no fine-tuning, no alignment, just the pretrained Mixture-of-Experts weights and enough JAX code to run them.

What Sets It Apart
  • Sparse by design: 8 experts with only 2 active per token means the 314B headline maps to a far smaller compute cost per forward pass — but you still need memory for all 314B weights resident at once.
  • Base, not instruct: this is the foundation checkpoint, so expect raw completion behavior, not a polite assistant. It's a substrate for your own fine-tuning, not a drop-in chatbot.
  • Genuinely permissive: Apache 2.0 covers both code and weights, so commercial use and redistribution are unencumbered — rarer than it sounds among "open" model releases.
  • Reference, not production: the authors explicitly note the MoE layer isn't optimized; it exists to prove the weights load and run correctly, not to serve traffic.
Who It's For

Great fit if you want a permissively licensed, large MoE base to study sparse architectures or build your own fine-tunes, and you have serious multi-GPU hardware (the weights run to hundreds of GB). Look elsewhere if you need a ready-to-use assistant, run on a single consumer GPU, or care about efficient serving — the code here is a correctness reference, and later open MoE releases like Mixtral, DeepSeek and Qwen offer far better size-to-quality tradeoffs.

Information

  • Websitegithub.com
  • OrganizationsxAI
  • AuthorsxAI (xai-org)
  • Published date2024/03/14

Categories

More Items

Enables RL post-training with million-token prompts under a fixed GPU budget by evaluating shared prompt state without autograd, retaining only minimal model state, and replaying short response branches; instantiated as GRPO and demonstrated on Qwen3.6-27B and GLM-5.2 up to multi-million token execution.

GitHub
AI Infra2026

Defines OpenTelemetry semantic conventions for generative AI telemetry — spans, metrics, and events for GenAI clients, the Model Context Protocol (MCP), and provider-specific integrations. Includes YAML models, human-readable docs, and reference implementations to standardize observability across GenAI deployments.

GitHub
AI Infra2024

Provides a lightweight build platform for HIP and ROCm that supports building ROCm, PyTorch, and JAX from source, multi-architecture nightly releases, and integrated CI/CD and developer tooling for Linux and Windows.