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.3.tar.gz (32.0 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.3-py3-none-any.whl (34.8 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: langtrain_ai-1.0.3.tar.gz
  • Upload date:
  • Size: 32.0 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.3.tar.gz
Algorithm Hash digest
SHA256 eae30f0d1e73d43b405e89a142156eda2e77f78a00194ce143460d539dce7af4
MD5 d546711062b87dd7dd500a79c6e6133d
BLAKE2b-256 b77a0d77e3937a94853b22e2ef1d9b7980829dfb480f4efa310aade40a72031c

See more details on using hashes here.

File details

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

File metadata

  • Download URL: langtrain_ai-1.0.3-py3-none-any.whl
  • Upload date:
  • Size: 34.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.3-py3-none-any.whl
Algorithm Hash digest
SHA256 62e1ded342fe15697fa25a9e0b86e8fcfd4ca153ea142c8f92966cd72a8d8036
MD5 094d999399ab4afc3e1f4d9ddb7cf744
BLAKE2b-256 e40cf527515f2a10a029a2604d305e702a1937bf6c2ce3dfa8cf747e1a6c30ee

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