domain-ml

Guide Rust ML development with constraints and tooling choices for inference and data pipelines.

Updated Jan 21, 2026
One-click install
npx skills add https://github.com/lywa1998/self-host-claude-marketplace --skill domain-ml-lywa1998
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: domain-ml
Source: https://github.com/lywa1998/self-host-claude-marketplace/tree/main/plugins/rust-skills/skills/domain-ml
Command: npx skills add https://github.com/lywa1998/self-host-claude-marketplace --skill domain-ml-lywa1998

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Building ML/AI applications in Rust often requires navigating complex constraints around memory efficiency, GPU acceleration, and model portability. This skill provides design guidance, tooling recommendations, and best practices to streamline development from data to deployment.

Core Features & Use Cases

  • Efficient memory management with zero-copy data handling and streaming data pipes.
  • GPU-accelerated inference using CUDA/Metal backends through libraries like tch-rs, candle, and burn.
  • Model portability via ONNX runtimes (ONNX), enabling easy deployment across Rust projects.
  • End-to-end pipelines that support lazy data loading, batching, and serving of predictions.

Quick Start

Create a minimal Rust project that loads an ONNX model and runs a batched inference to validate throughput.

Frequently Asked Questions about domain-ml

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

FAQPage Schema
How do I run ONNX model inference in Rust with GPU acceleration?

Rust ONNX inference leverages runtimes with CUDA or Metal backends via tch-rs, candle, or burn for GPU-accelerated execution. This approach enables fast, portable model deployment across CPU and GPU environments.

What is the best way to handle memory efficiently when building ML data pipelines in Rust?

Efficient ML data pipelines in Rust use zero-copy data handling and streaming pipes. This enforces memory-efficient data loading, batching, and serving of predictions without runtime overhead.

Can I use Rust for machine learning deployment across both CPU and GPU environments?

Yes, Rust ML deployment supports CPU and GPU environments. Libraries like ndarray handle CPU operations, while tch-rs, candle, and burn provide GPU backends like CUDA and Metal for accelerated inference.

How do I implement lazy initialization for standard model loading in Rust?

Implement lazy initialization for model loading in Rust to defer resource allocation until inference begins. This uses standard-model loading with libraries like candle or burn to optimize memory and startup performance.

Does Rust support deterministic behavior for machine learning inference engines?

Rust enables deterministic behavior for ML inference engines by enforcing strict memory handling and controlled execution flows. This ensures reproducible prediction outputs across standard model deployments.

When should I not use Rust for building machine learning inference engines?

Avoid building ML inference engines in Rust if your project requires rapid prototyping without strict memory constraints, or lacks compatible GPU backend support via CUDA or Metal for your specific target platform.