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.4.tar.gz (30.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.4-py3-none-any.whl (33.1 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: langtrain_ai-1.0.4.tar.gz
  • Upload date:
  • Size: 30.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.4.tar.gz
Algorithm Hash digest
SHA256 ad1dd647b094b70c84c109b450d804a2d00b9345679d68ee63e53ab770f91c57
MD5 47b91dabe28030168ffdf39a206e9c6f
BLAKE2b-256 4354f3e63a0526dd36d8fdb88749757a26097bc857dcfa100d9f5ca54c401eb3

See more details on using hashes here.

File details

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

File metadata

  • Download URL: langtrain_ai-1.0.4-py3-none-any.whl
  • Upload date:
  • Size: 33.1 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.4-py3-none-any.whl
Algorithm Hash digest
SHA256 a2c554dd6072cb888b386ed5404d3a6e4091d8a031b98c47fedc488963d9ac90
MD5 045881ab07404bec0cd9ace892a1dde1
BLAKE2b-256 9fa7e222a2b2f7e2f16dfc081b40a8a273f76271b06f08c3afcedc8e4a4270bc

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