jax-resnet
by n2cholas
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
AI summary
ResNet library
Provides implementations and checkpoints for various ResNet variants using JAX and Flax.
- stars
- 105
- forks
- 8
- watching
- 4
- 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.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
RL library
Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces.
Deep learning models
Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets.
ResNet implementation
An implementation of Residual Networks using Keras' functional API
Deep learning model
Represents an implementation of the Inception-ResNet v2 deep learning model in Keras.
ResNet variant
An open-source implementation of a res2net module using Keras and TensorFlow backend.
ResNet replication
Reproducing ResNet-v2 with MXNet on ImageNet and CIFAR10 datasets using Python.
Deep neural network framework
An implementation of ResNeXt models in Keras, allowing for efficient deep neural networks for image classification.
Graph library
A lightweight library for working with graph neural networks in jax.
Deep Learning Model
A deep learning model implementation of the DeepLab ResNet architecture for image segmentation tasks.
Deep Learning Library
A .NET Standard library providing C# bindings for the Apache MXNet deep learning framework
Deep networks
Implementations of deep learning architectures in Keras using the Functional API
Neural Network Library
An artificial neural network library for rapid prototyping and extension in Haskell.
Deep network architecture
An implementation of Wide Residual Networks in PyTorch for efficient deep learning on CIFAR10/100 datasets.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions jax-resnet.