ml-code-review

Review AI/ML experiment Python code for correctness and reproducibility.

Updated Aug 27, 2026
One-click install
npx skills add https://github.com/schmidtkk/skills --skill ml-code-review
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: ml-code-review
Source: https://github.com/schmidtkk/skills/tree/main/ml-code-review
Command: npx skills add https://github.com/schmidtkk/skills --skill ml-code-review

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Reviews AI/ML experiment Python code for correctness, reproducibility, and best practices across PyTorch, TensorFlow, and JAX. It helps identify issues in training loops, data handling, and architectural choices that can affect results and reproducibility.

Core Features & Use Cases

  • Automated ML code reviews focusing on reproducibility, training loop correctness, data handling, and architectural best practices for ML models including diffusion models, Transformers, GANs, and VAEs.
  • Supports PyTorch, TensorFlow, and JAX codebases with domain-aware checks and actionable remediation steps.
  • Generates structured reports with severity levels, concrete fixes, and example invocations to speed up debugging and auditing.

Quick Start

Provide a comprehensive ML code review for a given training script to ensure correctness, reproducibility, and adherence to best practices.

Frequently Asked Questions about ml-code-review

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

FAQPage Schema
How do I get actionable fixes for issues found during an ML experiment code review?

To get actionable fixes for issues found during an ML experiment code review, this Skill generates structured reports detailing severity levels, concrete remediation steps, and example invocations to speed up debugging and auditing.

How do I review ML training code for reproducibility and correctness?

To review ML training code for reproducibility and correctness, this Skill analyzes Python scripts to verify seed setting, determinism, training loop logic, and data handling, providing structured reports with actionable fixes.

Does this ML code review tool support PyTorch, TensorFlow, and JAX?

Yes, this ML code review tool supports PyTorch, TensorFlow, and JAX codebases, applying domain-aware checks for training loop correctness, reproducibility, and architectural best practices across these frameworks.

Can I check my diffusion model or Transformer implementation for architectural best practices?

Yes, you can check diffusion model, Transformer, GAN, and VAE implementations for architectural best practices, receiving actionable remediation steps and structured reports to ensure correctness and adherence to established guidelines.

What does a machine learning code review check for in data loaders and experiment scripts?

A machine learning code review checks data loaders and experiment scripts for data handling correctness, reproducibility issues like missing seeds, and training loop errors, delivering a structured report with severity levels and concrete fixes.

How do I get actionable fixes for issues found during an ML experiment code review?

To get actionable fixes for issues found during an ML experiment code review, this Skill generates structured reports detailing severity levels, concrete remediation steps, and example invocations to speed up debugging and auditing.