Extending JAX with CUDA
Monday 1 Feb 2021, 17:30
→
18:30
Europe/Zurich
Description
Recorded Meeting Video:
https://www.youtube.com/watch?v=NJHEPV4Etmg
17:30
→
18:30
Presentation/Discussion
17:30
Dan Foreman-Mackey - Extending JAX with CUDA
1h
Speakers
:
Daniel Foreman-Mackey
(
Flatiron Institute
)
,
Kyle Stuart Cranmer
(
New York University (US)
)
,
Matthew Feickert
(
Univ. Illinois at Urbana Champaign (US)
)
Dan Foreman-Mackey
GitHub repo: Extending JAX with custom C++ and CUDA code
jax-dfm-2021-02-01.pdf
YouTube Recording