tedwards2412 / ripple

Differentiable Gravitational Waveforms with JAX
53 stars 15 forks source link

Capability of using float16/float32 in waveforms #10

Open kazewong opened 1 year ago

kazewong commented 1 year ago

Tensor core in Nvidia GPU supposedly does wonder, but they need either float16 or float32 to work with. See if this help boost the performance

tedwards2412 commented 1 year ago

I've removed the default float64 requirement within ripple so we can start trialing this in Jim