earthkit.utils.array.converter.FromJaxConverter

class earthkit.utils.array.converter.FromJaxConverter(xp_target)

Bases: FromUnknownConverter

__init__(xp_target)

Methods

__init__(xp_target)
to(array, target_backend, **kwargs)
to_cupy(array, **kwargs)
to_jax(array, **kwargs)
to_numpy(array, **kwargs)
to_torch(array, **kwargs)
to_unknown(array, **kwargs)

to_cupy(array, **kwargs)

to_jax(array, **kwargs)

to_numpy(array, **kwargs)