rust-gpu-scheduling

Schedule ML GPU workloads within VRAM budgets using queues and dependency tracking.

1|Updated Dec 30, 2025
One-click install
npx skills add https://github.com/gar-ai/mallorn --skill rust-gpu-scheduling
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: rust-gpu-scheduling
Source: https://github.com/gar-ai/mallorn/tree/main/.claude/skills/rust-integration-gpu-scheduling
Command: npx skills add https://github.com/gar-ai/mallorn --skill rust-gpu-scheduling

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

VRAM-aware GPU scheduling for ML model orchestration, balancing memory budgets with workload demands to prevent OOM and reduce idle time.

Core Features & Use Cases

  • VRAM-aware scheduling with a safety margin to minimize OOM risks across multiple models (Whisper, VideoMAE, CLAP, Qwen3, DINOv3).
  • Work item queues with priority handling and dependency tracking to enforce correct execution order.
  • Model lifecycle management: track loaded/unloaded state and estimated load times to reduce costly reloads.
  • VRAM budgeting, per-model accounting, and simple semaphore-based limiting to cap concurrent workloads.
  • Guidelines and extensible architecture to integrate into ML orchestration pipelines.

Quick Start

Instantiate the GPUScheduler with your VRAM configuration and submit your first WorkItem to begin scheduling.

Frequently Asked Questions about rust-gpu-scheduling

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I prevent VRAM out of memory errors when running multiple ML pipelines?

VRAM-aware GPU scheduling prevents out of memory errors by applying per-model memory budgets and safety margins across multiple ML pipelines. It tracks loaded and unloaded model states to cap concurrent workloads within your available memory.

How does GPU scheduling handle task dependencies and priority queues for ML workloads?

GPU scheduling for ML workloads handles task dependencies and priority queues by tracking work item execution order and priority levels. Dependency tracking enforces correct sequential execution while priority queuing reorders tasks dynamically as VRAM resources change.

What is the best way to manage model lifecycle state to reduce GPU reload times?

Managing model lifecycle state reduces GPU reload times by tracking loaded and unloaded states alongside estimated load durations. This lifecycle management minimizes costly model reloads during multi-model pipeline orchestration by maintaining active state awareness.

Can I use VRAM budgeting to cap concurrent workloads across models like Whisper and Qwen3?

VRAM budgeting caps concurrent workloads across models like Whisper and Qwen3 using per-model memory accounting and semaphore-based limiting. This approach governs task execution by enforcing strict memory limits tailored to each specific model's requirements.

Does GPU scheduling support reordering ML tasks when VRAM resources change dynamically?

GPU scheduling supports reordering ML tasks when VRAM resources change dynamically by applying queue-based scheduling with dependency tracking. This dynamic reordering ensures workload execution continues efficiently without exceeding memory budgets during resource fluctuations.

What are the limitations of semaphore-based limiting for GPU memory management?

Semaphore-based limiting for GPU memory management relies on simplified concurrency capping rather than precise real-time VRAM allocation tracking. It governs task execution queues effectively but may require safety margins to account for allocation overhead during dynamic model loading.