domain-ml

Installation
Summary

Machine learning and AI applications in Rust with tensor operations, model inference, and GPU acceleration.

  • Covers tensor libraries (ndarray), inference frameworks (tract for ONNX, candle, burn), and PyTorch bindings (tch-rs) for training and deployment workflows
  • Emphasizes memory efficiency through zero-copy operations, GPU batching, and standard model formats (ONNX) for portability across Python and Rust
  • Provides design patterns for model loading with lazy initialization, batched inference for GPU throughput, and async data pipelines to prevent GPU idle time
  • Includes code examples for inference servers and batched prediction, plus common pitfalls like tensor cloning and per-request model loading
SKILL.md

Machine Learning Domain

Layer 3: Domain Constraints

Domain Constraints → Design Implications

Domain Rule Design Constraint Rust Implication
Large data Efficient memory Zero-copy, streaming
GPU acceleration CUDA/Metal support candle, tch-rs
Model portability Standard formats ONNX
Batch processing Throughput over latency Batched inference
Numerical precision Float handling ndarray, careful f32/f64
Reproducibility Deterministic Seeded random, versioning

Critical Constraints

Related skills

More from zhanghandong/rust-skills

Installs
500
GitHub Stars
1.1K
First Seen
Jan 20, 2026