File size: 447 Bytes
481ffb7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | """
Chess Challenge - LLM Training Exercise
Train a 1M parameter model to play chess using custom tokenizers and architectures.
"""
from src.tokenizer import ChessTokenizer
from src.model import ChessConfig, ChessForCausalLM
from src.utils import count_parameters, print_parameter_budget
__version__ = "0.1.0"
__all__ = [
"ChessTokenizer",
"ChessConfig",
"ChessForCausalLM",
"count_parameters",
"print_parameter_budget",
]
|