Can this skill help with Out of Memory (OOM) errors?

Absolutely. By implementing mixed precision, it significantly reduces the GPU memory footprint, often allowing you to double your batch size or train larger models.

What is mixed precision training?

Mixed precision training is a technique that uses both 16-bit and 32-bit floating-point types during model training to make it faster and use less memory while maintaining model accuracy.

Which frameworks are compatible with this skill?

The skill is designed to work with major ML frameworks including PyTorch, TensorFlow/Keras, and provides general guidance for scikit-learn workflows.

Does this skill support PyTorch?

Yes, it provides specific implementation patterns for PyTorch Automatic Mixed Precision (AMP), including the use of GradScaler for numerical stability.

Mixed Precision ML Trainer

Name: Mixed Precision ML Trainer
Author: jeremylongshore

byjeremylongshore

•

983

•

データサイエンスとML

Optimizes machine learning model training by implementing mixed-precision techniques to reduce memory usage and accelerate computation.

The Mixed Precision Trainer skill automates the implementation of FP16 and BF16 training workflows within Claude Code, helping developers balance numerical stability with computational efficiency. It provides specialized guidance for PyTorch (AMP), TensorFlow, and scikit-learn, generating production-ready code that minimizes memory overhead while maintaining model accuracy. This skill is ideal for data scientists and ML engineers looking to speed up training cycles and optimize resource utilization on modern GPU hardware.

主な機能

01Generation of production-ready boilerplate for FP16 and BF16 training loops

02Memory usage optimization for large-scale model training workflows

03983 GitHub stars

04Numerical stability validation to prevent gradient underflow and overflow

05Automated implementation of Automatic Mixed Precision (AMP) for PyTorch and TensorFlow

06Configuration guidance for hardware acceleration using NVIDIA Tensor Cores

ユースケース

01Migrating legacy full-precision training scripts to modern mixed-precision standards

02Accelerating training times for deep learning models on limited GPU hardware

03Reducing VRAM consumption to accommodate larger batch sizes or complex architectures

What are Skills?·How to Install

Install with 🐟 Skill.Fish

npx skillfish add jeremylongshore/claude-code-plugins-plus-skills mixed-precision-trainer

For use in Claude.ai and ChatGPT

Download Skill

主な機能

01Generation of production-ready boilerplate for FP16 and BF16 training loops

02Memory usage optimization for large-scale model training workflows

03983 GitHub stars

04Numerical stability validation to prevent gradient underflow and overflow

05Automated implementation of Automatic Mixed Precision (AMP) for PyTorch and TensorFlow

06Configuration guidance for hardware acceleration using NVIDIA Tensor Cores

ユースケース

01Migrating legacy full-precision training scripts to modern mixed-precision standards

02Accelerating training times for deep learning models on limited GPU hardware

03Reducing VRAM consumption to accommodate larger batch sizes or complex architectures

What are Skills?·How to Install

Install with 🐟 Skill.Fish

npx skillfish add jeremylongshore/claude-code-plugins-plus-skills mixed-precision-trainer

For use in Claude.ai and ChatGPT

Download Skill