official-stockfish/nnue-pytorch
A PyTorch-based trainer for NNUE neural network evaluation in Stockfish chess engine.

Velocity · 7d
+0.2
★ / day
Trend
→steady
star history
This repository provides tooling to train NNUE (Efficiently Updatable Neural Network) evaluation functions for the Stockfish chess engine. It uses PyTorch to optimize the neural network weights on chess position datasets, then converts the trained weights for use in the C++ Stockfish engine. The project supports GPU acceleration via CUDA, ROCm, and Apple MPS backends.