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.0.tar.gz (22.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.0-py3-none-any.whl (24.3 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: langtrain_ai-1.0.0.tar.gz
  • Upload date:
  • Size: 22.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.7

File hashes

Hashes for langtrain_ai-1.0.0.tar.gz
Algorithm Hash digest
SHA256 5250cd18c78d6b6105892bc17178d90e4da55acc9543bc2edd69a7e009ee87b7
MD5 84a471aa2873d92135bc3318e536ef0e
BLAKE2b-256 54af510bfafaca1f9aaad55e11ca22ed81d29ea315286e5a74f25c3ce383d98e

See more details on using hashes here.

File details

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

File metadata

  • Download URL: langtrain_ai-1.0.0-py3-none-any.whl
  • Upload date:
  • Size: 24.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.7

File hashes

Hashes for langtrain_ai-1.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 10b6c4d7e728bab0de5a5192ff816e6c8bed994178bec35cc229c70dfe483ec5
MD5 d3f98601c3604a23934ca3d1abf2d2da
BLAKE2b-256 a8172afe88476db635ca77c520b641966e17ae8f0125a2c5ea2ce9109e7404f8

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