chessnano

ChessNano

A compact autoregressive transformer that reads chess moves in Standard Algebraic Notation and predicts the next one. No board representation and no evaluation function — the model sees notation and nothing else.

Code: github.com/Kcbir/chessnano

Files

Architecture

Quantization

Weights are trained with quantization simulated in the forward pass. The TurboQuant path applies a Hadamard rotation, quantises each coordinate pair's angle to 128 levels, and stores a one-bit Johnson–Lindenstrauss residual sketch scaled by the least-squares coefficient. Training and export share one reconstruction function, so the simulated quantizer matches the exported one.

Usage

import chess
from inference.tot_inference import ChessNanoPlayer, TotConfig

player = ChessNanoPlayer("chessnano_deployed.pt", tot_cfg=TotConfig(depth=1))

board, history = chess.Board(), []
for san in ["e4", "e5", "Nf3"]:
    board.push_san(san)
    history.append(san)

print(player.predict_move(history, board))

Candidate moves are masked against python-chess's legal move list before sampling, so the decoder cannot emit an illegal move.

Licence

MIT