BijectorCalculator.jaxify()#

BijectorCalculator.jaxify()#

Enables JAX NumPy for the node component.

Return type

NodeComponent