extending-jax

extending-jax

https://github.com/dfm/extending-jax

Python379

3 months ago

Reviews

Extending JAX with custom C++ and CUDA code

Search similar apps

License

MIT License

Share

Extending JAX with custom C++ and CUDA code

Creator

dfm

Related apps

tinygp

The tiniest of Gaussian Process libraries

Python296mit

5 days ago