search

normal-computing / thermox

Exact OU processes with JAX
Apache License 2.0
24 stars 3 forks source link

Add A_spd to sample #5

Closed SamDuffield closed 4 months ago

SamDuffield commented 4 months ago

Yeah it doesn’t solve the GPU issue, but does give the user the ability to sample on GPU now.

I couldn’t work out how to do symmetry checking that is jit-compatible