Features

Encode Boards

Encode a chess board into a tensor.

notebooks/features/encode-boards.ipynb

Load Models

Load a model from a file or the Hugging Face Hub.

notebooks/features/load-models.ipynb

Move Prediction

Make a move prediction for a given chess board.

notebooks/features/move-prediction.ipynb

Run Models on GPU

Take advantage of GPU acceleration.

notebooks/features/run-models-on-gpu.ipynb

Evaluate Models on Puzzles

Evaluate a model on a set of puzzles.

notebooks/features/evaluate-models-on-puzzles.ipynb

Convert Official Weights

Convert lc0 networks to onnx.

notebooks/features/convert-official-weights.ipynb

Visualise Heatmaps

Visualise saliency heatmaps or encodings for a given chess board.

notebooks/features/visualise-heatmaps.ipynb

Probe Concepts

Probe the concepts with a dataset.

notebooks/features/probe-concepts.ipynb