search

hpc4cmb / toast

Time Ordered Astrophysics Scalable Tools
Other
44 stars 39 forks source link

Updating JAX #688

Closed nestordemeure closed 1 year ago

nestordemeure commented 1 year ago

replaced DeviceArray with Jax.Array to comply with the jax.Array migration.