extending-jax

Extending JAX with custom C++ and CUDA code

369 stars

10 watching

21 forks

Language: Python

last commit: about 1 month ago

Linked from 1 awesome list

cudajaxxla

Backlinks from these awesome lists:

n2cholas/awesome-jax