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.1.tar.gz (21.4 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.1-py3-none-any.whl (23.8 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: langtrain_ai-1.0.1.tar.gz
  • Upload date:
  • Size: 21.4 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.1.tar.gz
Algorithm Hash digest
SHA256 5b542e3771d79b486cc65b83ae56353c59327c9951a5b887f0f4b37c3db31a34
MD5 e6d67c6d103f751e6bc4826e9269bdbc
BLAKE2b-256 494bd69e5505dc85202989e86bc9dc9eec69731556c99896a00b03ee864615fc

See more details on using hashes here.

File details

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

File metadata

  • Download URL: langtrain_ai-1.0.1-py3-none-any.whl
  • Upload date:
  • Size: 23.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.1-py3-none-any.whl
Algorithm Hash digest
SHA256 225f38d7623489da056bd2914c1e89f8380570a4ebefb411f171ea4b5567a2ee
MD5 21cee9b0fa266739bbc3fcb357680b0d
BLAKE2b-256 ef417bf1c89765cb4f84eefad9ffa83173f8df4433a1d5b46cce85b02ca6f830

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