JAX ResNet - Implementation of ResNet, ResNet-D, and ResNeSt in Flax

A Flax (Linen) implementation of ResNet (He, Kaiming, et al. 2015), ResNet-D (He, Tong et al. 2020), and ResNest (Zhang, Hang et al. 2020). The code is modular so you can mix and match the various stem, residual, and bottleneck implementations.

Related Repos



szha KDD18 London: Deep Learning and Natural Language Processing with Apache MXNet (Incubating) Gluon Time: Tuesday, August 21, 2018Location: ICC Capital Suite Room 14+15+16 Presenter: Alex Smola, Leonard Lausen, Haibin Lin
 

taki0112 Simple Tensorflow implementation of Diverse Image-to-Image Translation via Disentangled Representations (ECCV 2018 Oral)
 

huguyuehuhu A PyTorch Reproduction of HCN Co-occurrence Feature Learning from Skeleton Data for Action Recognition and Detection with Hierarchical Aggregation. Chao Li, Qiaoyong Zhong, Di Xie, Shiliang Pu, IJCAI 2018. Arxiv Preprint
 

ikostrikov pytorch-flows A PyTorch implementations of Masked Autoregressive Flow and some other invertible transformations from Glow: Generative Flow with Invertible 1x1 Convolutions and Density estimation using Real NVP. For MAF, I'm gett
 

asyml Texar is a toolkit aiming to support a broad set of machine learning, especially natural language processing and text generation tasks. Texar provides a library of easy-to-use ML modules and functionalities for composing
 
Popular
3.2k

onnx Open Neural Network eXchange (ONNX) Model Zoo The ONNX Model Zoo is a collection of pre-trained models for state-of-the-art models in deep learning, available in the ONNX format. Accompanying each model are Jupyter n
 

aqeel13932 APES is a grid world environment with ability for partial observability. It could be used for multi-agent experiments. Its uniqueness is coming from the wide options for vision for every agent.