v2.0.0

2.0.0 (2026-03-27)

⚠ BREAKING CHANGES

• Add kv caching to AB-UPT (#131)
• simplify model init methods by moving config class instantiation to the config classes themself (#110)

✨ Features

• add attention mask support for padded token sequences (#115) (5474841)
• add cli tool to validate config first and lauch a slurm job via… (#94) (d734b6a)
• Add kv caching to AB-UPT (#131) (bfea608)
• add noether-init for scaffolding (#113) (de3b6b9)
• add signal handlers to BaseTrainer (#111) (5769509)
• add sphinx doctest (#100) (e3ef887)
• addnoether-init cli command to create a new boilerplate project to get started with implementing custom models and datasets. After running this command, the train command is printed, which the user can run to start a test training run immediately. (de3b6b9)
• eliminate requirement for custom root schema (#108) (81a62ae)
• match callback state dicts by key instead of position on resume (#114) (12bbaba)
• support for resuming training without data loading (#130) (492f041), closes #55

🐛 Bug Fixes

• bring back missing tutorial image (#112) (4772538)
• Remove bad lr scheduler default value and make it required instead (#129) (d4dbb0a)
• replace wrong dataset class for stats cli tool (#103) (4cad333)
• split input sequence correctly and enable using kv_dim (#116) (77dc41d)
• submit job CLI was not intercepting Hydra properly (#127) (9e624b8)
• variable duplication bug in caeml preprocessing script to st… (#125) (22ac5f0)
• various smaller bugs (#104) (308607a)
• wrong semantics for BestCheckpoint tolerance config (#105) (b5895cb)
• zero division in ProgressCallback in edge case (#122) (b3811b0)

♻️ Code Refactoring

• simplify model init methods by moving config class instantiation to the config classes themself (#110) (ae2ed95)