v1.2.1

New minor release with a variety of improvements relating to testing, documentation and development. Transition from torchtyping to jaxtyping is one the most significant changes.

What's Changed

• Replace torchtyping with jaxtyping by @dkamm in #171
• Run poetry lock by @rusheb in #178
• Add verbose flag to disable tqdm on model.generate(...) by @afspies in #185
• Make tracr plot show outside on colab by @ArthurConmy in #184
• Add positional_embedding_type to model properties table by @ckkissane in #176
• Run poetry lock --check in CI by @rusheb in #182
• Slice: doc and tests by @Xmaster6y in #166
• Separate tests into unit and acceptance tests by @rusheb in #191
• Grokking demo by @neelnanda-io in #193
• Configure coverage reports to measure branch coverage by @rusheb in #192
• Silence DeprecationWarning for distutils by @rusheb in #187
• Clone pos embed by @slavachalnev in #194
• Add pos embed hook tests by @slavachalnev in #196
• Fix test command in Readme by @valedan in #197
• Test constructor of FactoredMatrix by @rusheb in #188
• Add helper for logit attribution by @dkamm in #135
• issue and pr templates by @jbloomAus in #203

New Contributors

• @Xmaster6y made their first contribution in #166
• @slavachalnev made their first contribution in #194
• @valedan made their first contribution in #197

Full Changelog: v1.2...v1.2.1