jax-models
Unofficial JAX implementations of deep learning research papers
150 stars
5 watching
8 forks
Language: Python
last commit: over 2 years ago
Linked from 1 awesome list
artificial-intelligencecomputer-visionconvolutional-neural-networksdeep-learningflaxjaxmachine-learningresearch-paper-implementationtransformers
n2cholas/awesome-jax