rockpool.nn.modules.native.instant

Encapsulate a simple instantaneous function as a jax module

Classes

Instant(*args, **kwargs)

Wrap a callable function as an instantaneous Rockpool module

InstantJax(*args, **kwargs)

Wrap a callable function as an instantaneous Rockpool module, with a Jax backend

InstantMixin(shape, function, *args, **kwargs)

Wrap a callable function as an instantaneous Rockpool module

InstantTorch(*args, **kwargs)

Wrap a callable function as an instantaneous Rockpool module, with a Torch backend