← all repositories

official-stockfish/nnue-pytorch

A PyTorch-based trainer for NNUE neural network evaluation in Stockfish chess engine.

478 stars Python Domain AppsML Frameworks
nnue-pytorch
Not currently ranked — collecting fresh signals.
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.

Frequently asked

What is official-stockfish/nnue-pytorch?
A PyTorch-based trainer for NNUE neural network evaluation in Stockfish chess engine.
Is nnue-pytorch open source?
Yes — official-stockfish/nnue-pytorch is open source, released under the GPL-3.0 license.
What language is nnue-pytorch written in?
official-stockfish/nnue-pytorch is primarily written in Python.
How popular is nnue-pytorch?
official-stockfish/nnue-pytorch has 478 stars on GitHub.
Where can I find nnue-pytorch?
official-stockfish/nnue-pytorch is on GitHub at https://github.com/official-stockfish/nnue-pytorch.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.