kd.knn.convert

kauldron.klinen.convert(

cls: kauldron.klinen.convert_utils._ClsT,

) → kauldron.klinen.convert_utils._ClsT

Decorator that convert a flax class into klinen.

This is equivalent to inheriting from klinen.Module, but the new attributes/properties (.init_bind,…) are not detected by static type checker, so inheritance is prefered. This function allow quick usage on Colab.

Parameters:

cls – The flax module class

Returns:

The decorated klinen module class