search

hpc4cmb / toast

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

Small perf improvement for the JAX Healpix kernel #682

Closed nestordemeure closed 1 year ago

nestordemeure commented 1 year ago

Small perf improvement for the JAX Healpix kernel (get speedup to expected value when running in non-hybrid).