You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Ensuring that Stheno supports all of its functionality on the GPU (where appropriate) is quite important moving forwards. This will have to be CuArrays.jl-based. To achieve this, we must ensure that:
All base mean functions and kernels support GPU inputs
All operations on all compositions are implemented in terms of operations that CuArrays can support.
logpdf, rand, elbo, etc work correctly with CuArrays.
Zygote works efficiently on all of the above.
For the most part, I expect this functionality to require writing quite a number of tests, and making tweaks where necessary.
The text was updated successfully, but these errors were encountered:
Ensuring that Stheno supports all of its functionality on the GPU (where appropriate) is quite important moving forwards. This will have to be CuArrays.jl-based. To achieve this, we must ensure that:
logpdf
,rand
,elbo
, etc work correctly withCuArrays
.For the most part, I expect this functionality to require writing quite a number of tests, and making tweaks where necessary.
The text was updated successfully, but these errors were encountered: