Make your ML experiment wrapper scripts smarter with...

Install • Tutorial / Demo • Documentation • FAQs • Releases

🌟 Labtasker makes ML experiment wrapper scripts smarter with task prioritization, failure handling, halfway resume and more: just change 1 line of code.

If you like our project, please give us a star ⭐ on GitHub for latest update.

✨ When and Where to Use

TLDR: Replace for loops in your experiment wrapper script with labtasker to enable features like experiment parallelization, dynamic task prioritization, failure handling, halfway resume, and more.

🐳 For detailed examples and concepts, check out the documentation.

🧪️ A Quick Demo

This demo shows how to easily submit task arguments and run jobs in parallel.

It also features an event listener to monitor task execution in real-time and automate workflows, such as sending emails on task failure.

For more detailed steps, please refer to the content in the Tutorial / Demo.

⚡️ Features

• ⚙️ Easy configuration and setup.
• 🧩 Versatile and minimalistic design.
• 🔄 Supports both CLI and Python API for task scheduling.
• 🔌 Customizable plugin system.

🔮 Supercharge Your ML Experiments with Labtasker

• ⚡️ Effortless Parallelization: Distribute tasks across multiple GPU workers with just a few lines of code.
• 🛡️ Intelligent Failure Management: Automatically capture exceptions, retry failed tasks, and maintain detailed error logs.
• 🔄 Seamless Recovery: Resume failed experiments with a single command - no more scavenging through logs or directories.
• 🎯 Real-time Prioritization: Changed your mind about experiment settings? Instantly cancel, add, or reschedule tasks without disrupting existing ones.
• 🤖 Workflow Automation: Set up smart event triggers for email notifications or task workflow based on FSM transition events.
• 📊 Streamlined Logging: All stdout/stderr automatically organized in .labtasker/logs - zero configuration required.
• 🧩 Extensible Plugin System: Create custom command combinations or leverage community plugins to extend functionality.
• 🦾 AI Agent Skills: First-class skill definitions for Claude Code and OpenCode — let your AI assistant decompose and manage experiment scripts automatically.

🛠️ Installation

Note

You need a running Labtasker server to use the client tools. Simply use the installed Python CLI labtasker-server serve or use docker-compose to deploy the server. See deployment instructions.

1. Install via PyPI

# Install with optional bundled plugins
pip install 'labtasker[plugins]'

2. Install the Latest Version from GitHub

pip install git+https://github.com/luocfprime/labtasker.git

🚀 Quick Start

Use the following command to launch a labtasker server in the background:

labtasker-server serve &

Use the following command to quickly setup a labtasker queue for your project:

labtasker init

Then, use labtasker submit to submit tasks and use labtasker loop to run tasks across any number of workers.

Tip

Use AI to help decompose your experiment scripts. Install the Labtasker skill for your agent:

Claude Code — install via marketplace:

/plugin marketplace add luocfprime/labtasker
/plugin install labtasker-skill@labtasker

Or other agents — install via CLI:

npx skills add luocfprime/labtasker

Or copy skills/labtasker/SKILL.md to ~/.claude/skills/labtasker/SKILL.md

📚 Documentation

For detailed information on demo, tutorial, deployment, usage, please refer to the documentation.

🔒 License

See LICENSE for details.