flaxmodels
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
AI summary
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
- stars
- 240
- forks
- 24
- watching
- 6
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
ResNet library
Provides implementations and checkpoints for various ResNet variants using JAX and Flax.
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
Language model toolkit
A guide to using pre-trained large language models in source code analysis and generation
Deep learning model
Represents an implementation of the Inception-ResNet v2 deep learning model in Keras.
ML framework builder
A framework for training and serving machine learning models using JAX, with a focus on flexibility and customization.
ML models
Provides pre-trained machine learning models for natural language processing tasks using Clojure and the clj-djl framework.
google/paxml461
ML framework
A framework for configuring and running machine learning experiments on top of Jax.
Gradient optimizer library
A gradient processing and optimization library designed to facilitate research and productivity in machine learning by providing building blocks for custom optimizers and gradient processing components.
Model advisor
An LLM-based tool to assist deep learning model development
Deep Learning Model
A deep learning model implementation of the DeepLab ResNet architecture for image segmentation tasks.
deepset-ai/farm1.7K
Model adapter
An open-source framework for adapting representation models to various tasks and industries
Deep learning trainer
Provides a flexible and configurable framework for training deep learning models with PyTorch.
Language model
An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling
Graph library
A lightweight library for working with graph neural networks in jax.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions flaxmodels.