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 |
---|---|---|
| Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
| Provides implementations and checkpoints for various ResNet variants using JAX and Flax. | 105 |
| Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 467 |
| A guide to using pre-trained large language models in source code analysis and generation | 1,789 |
| Represents an implementation of the Inception-ResNet v2 deep learning model in Keras. | 180 |
| A framework for training and serving machine learning models using JAX, with a focus on flexibility and customization. | 212 |
| Provides pre-trained machine learning models for natural language processing tasks using Clojure and the clj-djl framework. | 0 |
| A framework for configuring and running machine learning experiments on top of Jax. | 461 |
| 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 |
| An LLM-based tool to assist deep learning model development | 28 |
| A deep learning model implementation of the DeepLab ResNet architecture for image segmentation tasks. | 602 |
| An open-source framework for adapting representation models to various tasks and industries | 1,743 |
| Provides a flexible and configurable framework for training deep learning models with PyTorch. | 1,196 |
| An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling | 245 |
| A lightweight library for working with graph neural networks in jax. | 1,380 |