Skip to content

Claude Code Machine Learning plugins and skills

Flowy lists 23 plugins in the Machine Learning category for Claude Code. training or serving models, neural networks, fine-tuning, ML frameworks. NOT "uses an LLM", which is true of every plugin here Every listing shows what is inside before you install, who built it, and whether it is Auto-invoked, meaning a FLOW.md router fires the right skill as you prompt instead of leaving them dormant.

pytorch

pytorch

Gives you a Python library for building and training deep learning models, with tensor math that runs on GPUs and a flexible autograd system for computing gradients. Work in familiar Python and reuse libraries such as NumPy and SciPy.

@pytorch@pytorchInstall Plugin
mlflow

mlflow

Skills for the MLflow repository that analyze failed GitHub Actions jobs, review pull requests, triage issues, and enforce Python and workflow coding conventions.

8 skills
@mlflow@mlflowInstall Plugin
skills

skills

Skills and an MCP server that let Claude Code work with the Hugging Face Hub, covering model training, dataset creation, the hf CLI, and deploying models to Amazon SageMaker.

26 skills
@huggingface@huggingfaceInstall Plugin
feast

feast

Gives a machine learning team one place to define features and serve them consistently, so training data and live predictions use the same values. Covers the offline store, online store and feature server.

4 skills
@feast-dev@feast-devInstall Plugin
civitai

civitai

Plugin for developers of the Civitai codebase, with skills that guide adding new AI model ecosystems, base models, generation support, prompt-enhancement guides, and training support to the app and its trainers.

52 skills
@civitai@civitaiInstall Plugin
model-optimizer

model-optimizer

Skills for NVIDIA Model Optimizer, a library for compressing and speeding up machine learning models through quantization, pruning, distillation, and neural architecture search. Includes skills for benchmarking model kernels and comparing evaluation results in MLflow.

16 skills
@nvidia@nvidiaInstall Plugin
openrag

openrag

Skills for installing, running locally, and integrating OpenRAG, a Retrieval-Augmented Generation tool for semantic document search and AI-powered conversations over your own documents.

3 skills
@langflow-ai@langflow-aiInstall Plugin
nemotron

nemotron

Skills for planning and chaining Nemotron model customization steps such as fine-tuning, RL alignment and pretraining, plus a reference desk for Nemotron Lightning and tools for adding models, steps and patterns.

13 skills
@nvidia-nemo@nvidia-nemoInstall Plugin
claude-vision-skill

claude-vision-skill

Gives Claude Code a way to describe and analyze images when the current model cannot read images directly. It sends the image to a vision-capable model and returns a text description, for images shared as a local path or URL.

1 skill
@asuojun@asuojunInstall Plugin
prime-rl

prime-rl

Trains large language models with reinforcement learning across many GPUs, asynchronously and at scale. Covers config setup, installation, running evals on checkpoints, the local run dashboard, and building CUDA kernels.

10 skills
@primeintellect-ai@primeintellect-aiInstall Plugin
ai-infra-auto-driven-skills

ai-infra-auto-driven-s…

Helps an agent run fair LLM serving benchmarks across SGLang, vLLM and TensorRT-LLM, read profiler traces down to the kernel, and estimate GPU memory and concurrency from startup logs. Ships with skills for model architecture diagrams too.

10 skills
@bbuf@bbufInstall Plugin
nvflare

nvflare

Skills for federated learning with NVIDIA FLARE, a Python SDK for adapting existing ML and deep learning workflows to federated training. Covers converting Hugging Face Trainer or SFTTrainer code, and running and reporting Auto-FL optimization of NVFlare jobs.

11 skills
@nvidia@nvidiaInstall Plugin
isaaclab-arena

isaaclab-arena

Lets robotics teams build simulated benchmark scenes and test robot policies in them at scale, without hand-writing a separate setup for each robot, object and scenario combination. Covers environment authoring, test runs and pulling evaluation results.

10 skills
@isaac-sim@isaac-simInstall Plugin
sparkrun

sparkrun

Launches, monitors and stops LLM inference workloads on one or more NVIDIA DGX Spark machines, so you can serve models across a cluster without Slurm or Kubernetes. Covers installing the tool and setting up the cluster, and managing the recipes that define each model run.

3 skills
@spark-arena@spark-arenaInstall Plugin
torchjd

torchjd

Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).

4 skills
@simplexlab@simplexlabInstall Plugin
karpathy-skill

karpathy-skill

A Claude Code skill that makes Claude act as a thinking advisor using Andrej Karpathy's mental models and communication style. It helps analyze AI reliability, learning methods, industry trends and product design from his perspective.

1 skill
@alchaincyf@alchaincyfInstall Plugin
dspy-agent-skills

dspy-agent-skills

Helps coding agents write, evaluate and optimize DSPy programs, from typed signatures and modules to metric-driven optimization and long-context reasoning. Covers the whole workflow so you can build and tune LLM pipelines without relearning the API each time.

5 skills
@intertwine@intertwineInstall Plugin
ovrtx

ovrtx

ovrtx is a C and Python library for embedding Omniverse RTX sensor simulation and visualization directly into applications.

39 skills
@nvidia-omniverse@nvidia-omniverseInstall Plugin
laya-playground

laya-playground

A website, three games, a benchmark and an agent skill for Laya, the open-source decision model: typed questions in, real probabilities out, one forward pass, no generated text. It runs on your own machine. The live site is at brainfunctioncollapse.com/laya.

1 skill
@wdobry@wdobryInstall Plugin
distil-cli-skill

distil-cli-skill

An Agent Skill for building task-specific small language models on the distil labs platform: from raw data or production traces, through teacher evaluation and synthetic data generation, to a finetuned, evaluated, deployable student model.

1 skill
@distil-labs@distil-labsInstall Plugin
building-with-typesafe-jev

building-with-typesafe…

Unofficial skill that teaches coding agents to build with TypeSafe AI's Jev: typed decisions, calibrated confidence, and prior art from 150+ community projects.

1 skill
@aaddrick@aaddrickInstall Plugin
opticalmodeler

opticalmodeler

English · 简体中文 · 日本語 / v2.0.0 Use Skill Cases Evidence A new view of the formal model. Apparatus layout and beam geometry retained; private instrument identities and scene files excluded. Design the path. Keep the evidence.

1 skill
@k-telux@k-teluxInstall Plugin
blatant-why

blatant-why

AI-powered biologics design campaign agent, multi-agent orchestration with BoltzGen, PXDesign, Protenix, and 200+ cloud tools. Antibodies, nanobodies, de novo binders, and beyond.

19 skills
@001tmf@001tmfInstall Plugin

FAQ

What are the best Claude Code Machine Learning plugins?
Flowy lists 23 plugins for Machine Learning, ranked by fit and signal rather than raw install count. Open any listing to read every skill in full before you install it.
Do these Machine Learning skills run automatically?
The Auto-invoked ones do. They ship a FLOW.md router and a hook, so Claude Code loads the matching skill as you prompt. Indexed listings install normally, and their skills fire only when Claude picks them from their descriptions.
How does Flowy decide a plugin is a Machine Learning plugin?
The tag is derived from the plugin's own content, its skill names, descriptions, README and file types, not from a tag someone typed. A plugin appears here only when its real content matches.
Are these Machine Learning plugins free?
Yes. Flowy is free and open source with nothing gated. You can read every skill before installing, and each listing links to its upstream repository and its creator.

Browse every plugin