v0.7.0

API changes

This version contains an API change to the kernelize function that makes it possible to use different kernels for inference/training/torch.compile. This requires a small adjustment to how kernelize is called, see the kernelize documentation for more information. In short, to kernelize a model for inference, use:

model = MyModel(...)
model = kernelize(model, mode=Mode.INFERENCE)

For training:

model = MyModel(...)
model = kernelize(model, mode=Mode.TRAINING)

What's Changed

• Add get_local_kernel function by @danieldk in #102
• Support registering inference/training-specific layers by @danieldk in #103
• Set version to 0.7.0.dev0 by @danieldk in #104

Full Changelog: v0.6.2...v0.7.0