Extending JAX with CUDA

Session

Presentation/Discussion

1 Feb 2021, 17:30

1. Dan Foreman-Mackey - Extending JAX with CUDA

Daniel Foreman-Mackey (Flatiron Institute), Kyle Stuart Cranmer (New York University (US)), Matthew Feickert (Univ. Illinois at Urbana Champaign (US))

01/02/2021, 17:30