Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
AlphaUncommon

dfm/extending-jax

Extending JAX with custom C++ and CUDA code

Owner
dfm (personal account)
Stars
Stars: 402
Forks
Forks: 23
Language
Python
License
MIT
Created
2021
Topics
jaxxlacuda
Tier
Alpha (250 to 699 ★) What this tier means
View on GitHub

More repos by dfm

All 8 repos by dfm

Nearby in Python

All Python repos