More CUDA compatibility - #52

Draft

zuhengxu wants to merge 16 commits into

mainfrom

Commits on Jun 20, 2025

start testing full gpu training for realnvp
zuhengxu
committed
789d9e0

Commits on Aug 21, 2026

Merge remote-tracking branch 'origin/main' into gpu
shravanngoswamii
committed
92a3ae3

Commits on Aug 28, 2026

Take the base distribution log-density on the device in the batched ELBO
shravanngoswamii
committed
98652be
Test the batched MvNormal log-density against Distributions
shravanngoswamii
committed
edde71b
Test the log-density on device arrays without a GPU
shravanngoswamii
committed
c87a81e
Extend the CUDA tests to the batched ELBO and planar flow training
shravanngoswamii
committed
bd1f9ec
Add a GPU planar flow demo with its own project
shravanngoswamii
committed
683077e
Correct the nsf note about running on the GPU
shravanngoswamii
committed
a8a9cac
Add a changelog entry for the GPU batched ELBO
shravanngoswamii
committed
0f53834
Bump patch version to 0.4.1
shravanngoswamii
committed
2e35568
Move the planar layer to the host structurally in the CUDA reference check
shravanngoswamii
committed
1943d77
State the observed Mooncake CUDA failure instead of the assumed cause
shravanngoswamii
committed
d190e52
Give Mooncake its own zero-derivative rule for the device draw
shravanngoswamii
committed
1805b2a
Train with Mooncake as well as Zygote in the CUDA tests
shravanngoswamii
committed
4f094f4
Use Mooncake in the GPU demo
shravanngoswamii
committed
523f113
Sum the mapped array so the GPU rule set covers it
shravanngoswamii
committed
122b49f