Skip to main content

AXIOM - World's most memory-efficient drift-free optimizer for PyTorch

Project description

QuarterBit AXIOM

World's Most Memory-Efficient Drift-Free Optimizer

90%+ memory savings. Zero precision loss. Train larger models.

The Problem

Training large AI models requires massive GPU memory for optimizer states. Adam stores 8 bytes per parameter just for momentum and variance - that's 8GB for a 1B parameter model.

Additionally, long training runs suffer from floating-point drift - tiny gradient updates accumulate rounding errors over millions of steps, causing:

  • Stalled convergence in late training
  • Numerical instability
  • Suboptimal final models

The Solution

QuarterBit AXIOM solves both problems:

Metric Adam AXIOM Improvement
Memory per param 8.0 bytes 0.76 bytes 90%+ savings
Precision drift Accumulates errors Drift-free Eliminated
Max model (16GB GPU) ~1.0B params ~2.7B params 2.7x larger

Installation

pip install quarterbit

Supported GPUs:

  • NVIDIA T4, V100, A100, L4, L40
  • NVIDIA RTX 30 series (3060-3090)
  • NVIDIA RTX 40 series (4060-4090)
  • NVIDIA H100, H200

Quick Start

from quarterbit import Axiom

# Drop-in replacement for Adam
optimizer = Axiom(model.parameters(), lr=1e-3)

# Optional: set training schedule for adaptive LR
optimizer.set_schedule(total_steps=10000, warmup_steps=1000)

# Train as usual
for batch in dataloader:
    loss = model(batch)
    loss.backward()
    optimizer.step()
    optimizer.zero_grad()

Why AXIOM?

1. Massive Memory Savings

Train 2.7x larger models on the same GPU. AXIOM uses proprietary compression to reduce optimizer state from 8 bytes to under 1 byte per parameter.

2. Drift-Free Training

Proprietary precision algorithms eliminate floating-point accumulation errors. Your model trains with perfect numerical stability from step 1 to step 1,000,000+.

3. Cloud Cost Reduction

Less memory = fewer GPUs = lower costs. Typical savings of 30-50% on cloud training bills.

4. Drop-In Replacement

Same API as PyTorch Adam. Change one line of code.

Benchmarks

Validated on GPT-2 (124M parameters):

Optimizer Final Loss Memory Status
AdamW 3.12 1,024 MB Baseline
8-bit Adam 3.14 640 MB -37%
Adafactor 3.18 512 MB -50%
AXIOM 3.11 95 MB -91%

See our Kaggle benchmark for full results including OOM stress tests and cost analysis.

Requirements

  • Python 3.8+
  • PyTorch 1.8+
  • NVIDIA GPU with CUDA support
  • Linux or Windows

Pricing

Tier Price Use Case
Free $0 Personal, research, evaluation
Pro $299/mo Commercial use, up to 10 GPUs
Team $2,499/mo Up to 100 GPUs, priority support
Enterprise Custom Unlimited, custom SLA

See quarterbit.dev/pricing for details.

License

Proprietary - see LICENSE for details. Free tier available.

Links


Copyright (c) 2026 Clouthier Simulation Labs. All rights reserved.

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

quarterbit-4.0.2.tar.gz (1.4 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

quarterbit-4.0.2-cp311-cp311-win_amd64.whl (1.5 MB view details)

Uploaded CPython 3.11Windows x86-64

File details

Details for the file quarterbit-4.0.2.tar.gz.

File metadata

  • Download URL: quarterbit-4.0.2.tar.gz
  • Upload date:
  • Size: 1.4 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.0

File hashes

Hashes for quarterbit-4.0.2.tar.gz
Algorithm Hash digest
SHA256 f8454daf186439caa24a8c764deb62ebdd66de4d3eb8bbe54c4e1637d93c3b41
MD5 24c4f2e72b06a5fc9bdea272c2d9064f
BLAKE2b-256 4525f2d7999f5db1c325af13909a47f20c5ba5326f9e8e055ba357613690fc68

See more details on using hashes here.

File details

Details for the file quarterbit-4.0.2-cp311-cp311-win_amd64.whl.

File metadata

  • Download URL: quarterbit-4.0.2-cp311-cp311-win_amd64.whl
  • Upload date:
  • Size: 1.5 MB
  • Tags: CPython 3.11, Windows x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.0

File hashes

Hashes for quarterbit-4.0.2-cp311-cp311-win_amd64.whl
Algorithm Hash digest
SHA256 9de3f378272294c7e11050fecab0ae2e467c73cb21c3344e83e2db36c43baa1a
MD5 053e3280a9a0147160ab0d122a4bbb51
BLAKE2b-256 ec2fe3f20555b77c1722efef157d783ce216d41a48a419083e8314a3ce46ed87

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page