flaxmodels
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
240 stars
6 watching
24 forks
Language: Python
last commit: over 1 year ago
Linked from 1 awesome list
deep-learningflaxgpt2jaxpretrained-modelsresnetstylegan2vgg
Related projects:
| Repository | Description | Stars |
|---|---|---|
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 n2cholas/jax-resnet | Provides implementations and checkpoints for various ResNet variants using JAX and Flax. | 105 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 467 |
 vhellendoorn/code-lms | A guide to using pre-trained large language models in source code analysis and generation | 1,789 |
 yuyang-huang/keras-inception-resnet-v2 | Represents an implementation of the Inception-ResNet v2 deep learning model in Keras. | 180 |
 erfanzar/easydel | A framework for training and serving machine learning models using JAX, with a focus on flexibility and customization. | 212 |
 scicloj/scicloj.ml.clj-djl | Provides pre-trained machine learning models for natural language processing tasks using Clojure and the clj-djl framework. | 0 |
 google/paxml | A framework for configuring and running machine learning experiments on top of Jax. | 461 |
 google-deepmind/optax | 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. | 1,730 |
 fabprezja/keras-gpt-copilot | An LLM-based tool to assist deep learning model development | 28 |
 isht7/pytorch-deeplab-resnet | A deep learning model implementation of the DeepLab ResNet architecture for image segmentation tasks. | 602 |
 deepset-ai/farm | An open-source framework for adapting representation models to various tasks and industries | 1,743 |
 open-mmlab/mmengine | Provides a flexible and configurable framework for training deep learning models with PyTorch. | 1,196 |
 l0sg/relational-rnn-pytorch | An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling | 245 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,380 |