Skip to main content

The unified Python SDK for training, aligning, and deploying LLMs — text and vision.

Project description

Langtrain

The unified Python SDK for training, aligning, and deploying LLMs

PyPI License Docs


pip install langtrain-ai              # cloud API + dataset intelligence
pip install langtrain-ai[train]       # + local GPU training (text LLMs)
pip install langtrain-ai[vision]      # + vision LLMs (LLaVA, Qwen-VL, …)
pip install langtrain-ai[all]         # everything

What's inside

Module What it does
FastLanguageModel Unsloth-style API for text LLMs — local or cloud
FastVisionModel Same API for vision LLMs (LLaVA, Qwen-VL, InternVL, …)
AdaptiveRankTrainer Novel algorithm: SpectralLoRA + dynamic rank + TurboQuant
DatasetIntelligence Drop any file → auto model + training config
LangtrainClient Cloud API: fine-tune, deploy, chat, GPU, models
lt CLI lt login, lt fine-tune, lt analyze, lt gpu

Quick start

Drop any dataset — get a training config

from langtrain import DatasetIntelligence

report = DatasetIntelligence.analyze("my_data.jsonl")
report.print_summary()
# ┌─────────────────────────────────────────────┐
# │  Task type    instruction  (87% confidence) │
# │  Domain       medical                       │
# │  Samples      2,400                         │
# │  Model        meta-llama/Llama-3.1-8B       │
# │  Method       adaptive_rank  rank=16        │
# │  TurboQuant   ✓ polar_quant+qjl             │
# └─────────────────────────────────────────────┘

Local training with AdaptiveRank

from langtrain import FastLanguageModel, AdaptiveRankConfig

config = AdaptiveRankConfig(
    initial_rank=16,
    max_rank=64,           # grows/shrinks based on gradient variance
    use_turboquant_kv=True # PolarQuant 3-bit KV cache
)

model, tokenizer = FastLanguageModel.from_pretrained(
    "meta-llama/Llama-3.1-8B",
    load_in_4bit=True,
)
model = FastLanguageModel.get_peft_model(model, method="adaptive_rank", config=config)
FastLanguageModel.train(model, tokenizer, dataset, output_dir="./output")

Cloud API

from langtrain import LangtrainClient

client = LangtrainClient(api_key="lt_...")

# Check account + GPU options
print(client.me())
print(client.gpu.available())

# Fine-tune
job = client.fine_tune(
    model="meta-llama/Llama-3.1-8B",
    dataset_id="ds_xyz",
    method="adaptive_rank",
)
for step in job.stream():
    print(step)

CLI

lt login                              # authenticate
lt whoami                             # account + GPU availability
lt gpu                                # list GPU options
lt analyze my_data.jsonl              # dataset intelligence
lt fine-tune llama-3.1-8b data.jsonl  # launch training
lt jobs                               # list jobs
lt models                             # list models

Why Langtrain vs Unsloth?

Unsloth Langtrain
Text LLMs ✓ FastLanguageModel
Vision LLMs ✓ FastVisionModel
Training algorithm vanilla QLoRA AdaptiveRank (SpectralLoRA + dynamic rank)
Dataset analysis DatasetIntelligence (7-pass, auto model pick)
KV cache compression TurboQuant (6× memory, 8× speed)
Cloud training ✓ langtrain-server (A100/H100)
RL alignment ✓ DPO / GRPO / PPO / Constitutional AI
CLI lt fine-tune, lt analyze, lt gpu

Made by Langtrain AI · Docs · Discord

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

langtrain_ai-1.0.2.tar.gz (34.5 kB view details)

Uploaded Source

Built Distribution

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

langtrain_ai-1.0.2-py3-none-any.whl (37.8 kB view details)

Uploaded Python 3

File details

Details for the file langtrain_ai-1.0.2.tar.gz.

File metadata

  • Download URL: langtrain_ai-1.0.2.tar.gz
  • Upload date:
  • Size: 34.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.2

File hashes

Hashes for langtrain_ai-1.0.2.tar.gz
Algorithm Hash digest
SHA256 496a52a451649eb0f7bdd4ac021f5902bd731a2ac3381caadde2893c1d8f88fc
MD5 4fee05815d34f1b780001ac9122ef04b
BLAKE2b-256 2bd9cac5e90753df56d17ec9ea06dc13cbdc2babbb80c0d0b521104af05d03a3

See more details on using hashes here.

File details

Details for the file langtrain_ai-1.0.2-py3-none-any.whl.

File metadata

  • Download URL: langtrain_ai-1.0.2-py3-none-any.whl
  • Upload date:
  • Size: 37.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.2

File hashes

Hashes for langtrain_ai-1.0.2-py3-none-any.whl
Algorithm Hash digest
SHA256 3724c0c3dc6df68633c4389d2cfe740f94c83d33f282f71fb46c224a10b4f5ad
MD5 80afa31d63b67c4ae74fd672486edb8c
BLAKE2b-256 b7624675f423148e704f399f9613b829ba2c2a9431d4377d01a1b7b85161eeb7

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