nifty8.nifty2jax module#

convert(nifty_obj: ~nifty8.operators.operator.Operator | ~nifty8.domain_tuple.DomainTuple | ~nifty8.multi_domain.MultiDomain, dtype=<class 'float'>) → Model | Vector | ShapeWithDtype | Dict[str, ShapeWithDtype][source]#

shapewithdtype_from_domain(domain, dtype) → ShapeWithDtype | Dict[str, ShapeWithDtype][source]#

spaces_to_axes(domain, spaces) → Tuple | int | None[source]#: Converts spaces in a domain to axes of the underlying NumPy array.

wrap_nifty_call(op, target_dtype=<class 'float'>) → Callable[[Any], Vector][source]#