dfm/extending-jax

Extending JAX with custom C++ and CUDA code

4 best points · 0 comments · 1 Hacker News post · 402 GitHub stars · View on GitHub

Hacker News discussions

• Extending JAX with custom C++ and CUDA code (4 points, 0 comments, by cl3misch, 2021-01-12)