jax-resnet
ResNet library
Provides implementations and checkpoints for various ResNet variants using JAX and Flax.
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
104 stars
4 watching
8 forks
Language: Python
last commit: over 2 years ago
Linked from 1 awesome list
computer-visiondeep-learningflaxjaxmachine-learningneural-networksresnet
Related projects:
| Repository | Description | Stars |
|---|---|---|
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 238 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 ikostrikov/jaxrl | Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces. | 630 |
 4uiiurz1/pytorch-res2net | Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |
 raghakot/keras-resnet | An implementation of Residual Networks using Keras' functional API | 1,386 |
 yuyang-huang/keras-inception-resnet-v2 | Represents an implementation of the Inception-ResNet v2 deep learning model in Keras. | 180 |
 fupiao1998/res2net-keras | An open-source implementation of a res2net module using Keras and TensorFlow backend. | 16 |
 tornadomeet/resnet | Reproducing ResNet-v2 with MXNet on ImageNet and CIFAR10 datasets using Python. | 556 |
 titu1994/keras-resnext | An implementation of ResNeXt models in Keras, allowing for efficient deep neural networks for image classification. | 224 |
 google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,375 |
 isht7/pytorch-deeplab-resnet | A deep learning model implementation of the DeepLab ResNet architecture for image segmentation tasks. | 602 |
 deepakkumar1984/mxnet.sharp | A .NET Standard library providing C# bindings for the Apache MXNet deep learning framework | 149 |
| titu1994/inception-v4 | Implementations of deep learning architectures in Keras using the Functional API | 386 |
 jbarrow/lambdanet | An artificial neural network library for rapid prototyping and extension in Haskell. | 377 |
 xternalz/wideresnet-pytorch | An implementation of Wide Residual Networks in PyTorch for efficient deep learning on CIFAR10/100 datasets. | 333 |