quaxed.scipy¤
quaxed.scipy
¤
Quaxed jax.scipy.
This module wraps the functions in jax.scipy with quax.quaxify.
__all__ = ('linalg', 'special')
module-attribute
¤
Built-in immutable sequence.
If no argument is given, the constructor returns an empty tuple. If iterable is specified the tuple is initialized from iterable's items.
If the argument is a tuple, the return value is the same object.
quaxed.scipy.linalg
¤
Quaxed jax.scipy.linalg.
This module wraps the functions in jax.scipy.linalg with quax.quaxify. The
wrapping happens dynamically through a module-level __dir__ and
__getattr__. The list of available functions is in __all__ and
documented in the built-in jax.scipy.linalg library.
__all__ = ('svd',)
module-attribute
¤
Built-in immutable sequence.
If no argument is given, the constructor returns an empty tuple. If iterable is specified the tuple is initialized from iterable's items.
If the argument is a tuple, the return value is the same object.
quaxed.scipy.special
¤
Quaxed jax.scipy.special.
This module wraps the functions in jax.scipy.special with quax.quaxify. The
wrapping happens dynamically through a module-level __dir__ and
__getattr__. The list of available functions is in __all__ and
documented in the built-in jax.scipy.special library.
__all__ = ('bernoulli', 'betainc', 'betaln', 'beta', 'bessel_jn', 'digamma', 'entr', 'erf', 'erfc', 'erfinv', 'exp1', 'expi', 'expit', 'expn', 'factorial', 'gammainc', 'gammaincc', 'gammaln', 'gamma', 'i0', 'i0e', 'i1', 'i1e', 'logit', 'logsumexp', 'lpmn', 'lpmn_values', 'multigammaln', 'log_ndtr', 'ndtr', 'ndtri', 'polygamma', 'spence', 'sph_harm', 'xlogy', 'xlog1py', 'zeta', 'kl_div', 'rel_entr', 'poch', 'hyp1f1')
module-attribute
¤
Built-in immutable sequence.
If no argument is given, the constructor returns an empty tuple. If iterable is specified the tuple is initialized from iterable's items.
If the argument is a tuple, the return value is the same object.