brsx 0.2.0

AutoForge — automatic model building, training, and fine-tuning that calibrates to your hardware

brsx

AutoForge — build, train, and fine-tune models that calibrate themselves to your hardware.

brsx probes your GPU/CPU by actually running a few steps, picks a model size that fits your memory and speed target, and trains it. No config wrestling — pick a mode and go.

Install

pip install brsx

Optional extras (installed only if you need them):

pip install brsx[finetune]   # HuggingFace fine-tuning (transformers, peft, accelerate)
pip install brsx[qlora]      # 4-bit QLoRA (bitsandbytes, Linux/CUDA)
pip install brsx[tokenizer]  # fast BPE for the MTP branch
pip install brsx[all]        # everything

Usage

from brsx import automodel

automodel.run()

You'll get a menu:

1) Transformer training        — train a standard Transformer from scratch
2) MTP Transformer training    — Multi-Token Prediction heads (DeepSeek-V3 style)
3) HuggingFace model fine-tune — fine-tune a HF model (local path or hub id)
4) brsx (.pt) model fine-tune  — fine-tune an existing brsx model

Pick one and follow the prompts — everything (mode, learning rate, steps, checkpointing, data source) is asked interactively.

License

Apache-2.0. See LICENSE.