jax-models (pypi): Unofficial JAX implementations of deep learning research papers
Keywords: jax|flax|models|computer-vision|deep-learning|transformers|artificial-intelligence|convolutional-neural-networks|machine-learning|research-paper-implementation