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)[source]

Convert a series-like object to a JAX array.

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

Create a GWexpy object from a JAX array.