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.3.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.3-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.3.tar.gz.

File metadata

  • Download URL: quarterbit-4.0.3.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.3.tar.gz
Algorithm Hash digest
SHA256 b2cfa756f539a841ecc27a739730e610ab3272c5f428ae9f534d0bad944557fc
MD5 73aad57d8ba8db429120fab671c95c0b
BLAKE2b-256 470bfd20c4157dad408aaa383bc53ca6c8e50ac768d89404ecf288ac8503deb9

See more details on using hashes here.

File details

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

File metadata

  • Download URL: quarterbit-4.0.3-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.3-cp311-cp311-win_amd64.whl
Algorithm Hash digest
SHA256 6312f986367b8ac357dcfd4d1589b28380c212a414e208f77670bda39c2b4841
MD5 09c2a4ff04a44ec023e11ff3d1b08a17
BLAKE2b-256 1d430370de4ad19b3c6c730dd97cfc8d64e02bcc3790b40db45dcc2d0f93cc3f

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