Skip to content

[haiku]

Haiku models interfacing tools.

This submodule provides with a generic interface to wrap up any Haiku module instance that is to be trained through gradient descent.

This module exposes: * HaikuModel: Model subclass to wrap haiku.Model objects * JaxNumpyVector: Vector subclass to wrap jax.numpy.ndarray objects