Development documentation · 0.2.3 30c2f8ba · Intro examples tested with 0.2.3 · Version details · Known limitations

gwexpy.interop.jax_

JAX interop helpers.

Functions

from_jax(cls, array, t0, dt[, unit])	Create a GWexpy object from a JAX array.
to_jax(ts[, dtype])	Convert a series-like object to a JAX array.

gwexpy.interop.jax_.to_jax(ts, dtype=None)

Convert a series-like object to a JAX array.

gwexpy.interop.jax_.from_jax(cls, array, t0, dt, unit=None)

Create a GWexpy object from a JAX array.