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.