Skip to main content

Preflight-check whether a machine can train an AI workload before the run starts.

Project description

TuneFit

TuneFit is the preflight check for AI training.

TuneFit predicts whether a local machine can train a target AI workload before the real run starts.

It is built for the point before a profiler helps and before a failed training run wastes time.

What TuneFit Does

  • scans the local machine and runtime
  • estimates whether a workload is trainable on that machine
  • predicts peak VRAM, host RAM, and rough step time
  • returns PASS, WARN, or FAIL
  • points to likely bottlenecks
  • suggests the next change to try

Why It Exists

Many training runs fail too late:

  • out-of-memory on the first step
  • unsupported CUDA, MPS, or precision combinations
  • framework and driver mismatches
  • technically valid runs that are too slow to be useful

TuneFit moves that answer forward.

Quick Example

tunefit scan --output system.json
tunefit estimate --spec workload.yaml --system system.json

Typical output:

TuneFit verdict: PASS
Target: mps on Apple Silicon Integrated GPU
Peak VRAM: 824.0 MB
Peak RAM: 563.2 MB
Step time: 215.4 ms
Throughput: 1188.59 tokens/sec

Current Scope

  • CLI-first
  • local-only
  • CPU, NVIDIA CUDA, and Apple MPS
  • PyTorch-first validation
  • CNN and transformer workload templates

Commands

  • tunefit scan
  • tunefit estimate
  • tunefit validate
  • tunefit calibration list
  • tunefit calibration prune

Source And Documentation

Notes

TuneFit is intentionally lightweight and heuristic. It is designed to improve preflight decisions, not to replace a real profiler or guarantee exact runtime behavior for arbitrary custom training code.

Project details


Download files

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

Source Distribution

tunefit-0.1.0.tar.gz (37.9 kB view details)

Uploaded Source

Built Distribution

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

tunefit-0.1.0-py3-none-any.whl (36.2 kB view details)

Uploaded Python 3

File details

Details for the file tunefit-0.1.0.tar.gz.

File metadata

  • Download URL: tunefit-0.1.0.tar.gz
  • Upload date:
  • Size: 37.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.3

File hashes

Hashes for tunefit-0.1.0.tar.gz
Algorithm Hash digest
SHA256 daa8c50ce06095b3fa3f1ac8725ab52eed8d9b68786d7140730c78ec8edefe5b
MD5 6348c902e25ec35d7cbdf844ec25f7e3
BLAKE2b-256 f4bf6ddef738ce9011a2d607d6ee184be77d42626c24daa0a67d2ec2424d6923

See more details on using hashes here.

File details

Details for the file tunefit-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: tunefit-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 36.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.3

File hashes

Hashes for tunefit-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 62f584546be1a4cfcb50f5dc0890a1cac10a164fccf4a8a80cfced9231976b27
MD5 d1b2c1db42e6562e5b9d5eff3812cf6b
BLAKE2b-256 142cb084253987850dfa64b245b2a1236fa4b17b8ee8f1d9fba0e0a41a132e86

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