dogear

enter for all results · esc to close

Extending JAX with custom C++ and CUDA code by Dan Foreman-Mackey

github.comtool

Tutorial demonstrating the infrastructure required to provide custom ops in JAX.

visit github.com ↗
0
sign in to dogear this
from
JAX
added
2026-10-10
likes
0

JAX › Tutorials and Blog Posts: “Tutorial demonstrating the infrastructure required to provide custom ops in JAX.”