search

hpc4cmb / toast

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

Jax templates #726

Closed nestordemeure closed 8 months ago

nestordemeure commented 8 months ago

Started using the amplitude_flags input in the JAX kernels.

tskisner commented 8 months ago

Actually, @nestordemeure and I discussed a remaining issue offline. Will push an update shortly.