ai-agent

LLM Foundry

Name: LLM Foundry
Author: mosaicml

LLM training code for Databricks foundation models

4,438 stars591 forksPythonUpdated 3/25/2026100% free · open source

What it does

LLM-Foundry is a Python library for training and fine-tuning large language models on Databricks, allowing founders to build custom AI models for their startups

When to use it

•When you need to train a custom language model for a specific industry or domain
•When you want to fine-tune a pre-trained language model for your startup's use case
•When you require a scalable and efficient way to train large language models on a cloud-based platform like Databricks

Ready-to-paste prompt

python train.py --model-name transformer-xl --dataset-name wiki_text --batch-size 16 --learning-rate 1e-4 --num-workers 4

Heads up: Make sure you have a Databricks account and a cluster set up with the necessary dependencies installed, including the `databricks` library, before attempting to use LLM-Foundry

Saves to your device

Use with Claude

New

Skip the builder — one click puts this in Claude, Cursor, Antigravity and more.

✅ Light setup

Installs with a command or two; your AI agent can do it for you.

Try it instantly — no install

Claude Code

mkdir -p ~/.claude/skills/llm-foundry && curl -fsSL https://workflowstacks.com/api/skills/llm-foundry/claude-skill -o ~/.claude/skills/llm-foundry/SKILL.md

Open in another AI app

Opens the app with this repo with the prompt ready to go — no copy-paste needed.

Connect the whole catalog (MCP)

claude mcp add --transport http workflowstacks https://workflowstacks.com/api/mcp

Adds a WorkflowStacks connector to Claude Code: search and load any skill here by chatting.

Use with Claude

New

Skip the builder — one click puts this in Claude, Cursor, Antigravity and more.

✅ Light setup

Installs with a command or two; your AI agent can do it for you.

Try it instantly — no install

Claude Code

mkdir -p ~/.claude/skills/llm-foundry && curl -fsSL https://workflowstacks.com/api/skills/llm-foundry/claude-skill -o ~/.claude/skills/llm-foundry/SKILL.md

Open in another AI app

Opens the app with this repo with the prompt ready to go — no copy-paste needed.

Connect the whole catalog (MCP)

claude mcp add --transport http workflowstacks https://workflowstacks.com/api/mcp

Adds a WorkflowStacks connector to Claude Code: search and load any skill here by chatting.

Quick Actions

Details

Creator

mosaicml

Language

Python