jax-models
Unofficial JAX implementations of deep learning research papers
AI summary
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
- stars
- 151
- forks
- 8
- watching
- 5
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
Graph library
A lightweight library for working with graph neural networks in jax.
Deep learning model
Represents an implementation of the Inception-ResNet v2 deep learning model in Keras.
Deep Learning Tutorials
A collection of tutorials and resources on implementing deep learning models using Python libraries such as Keras and Lasagne.
Vision-Language Model
A multimodal AI model that enables real-world vision-language understanding applications
Deep learning framework
Implementations of various deep learning algorithms in Python using Theano and Lasagne.
ResNet library
Provides implementations and checkpoints for various ResNet variants using JAX and Flax.
RL library
Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces.
Deep learning optimizer
An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning.
Deep learning library
A collection of GPU-accelerated deep learning algorithms implemented in Python
ML models
Provides pre-trained machine learning models for natural language processing tasks using Clojure and the clj-djl framework.
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
NLP models
A collection of pre-trained natural language processing models
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
Language model
An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling
Featured in 1 awesome list
Each link jumps to the spot where the list mentions jax-models.