rohan-insitro's Stars
catalyst-team/catalyst
Accelerated deep learning R&D
nightrome/cocostuff
The official homepage of the COCO-Stuff dataset.
RobRomijnders/segm
Simple Semantic Segmentation
huggingface/pytorch-image-models
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
yasserius/mini-mnist-1000
A small subset of 1000 images from MNIST
rolandgvc/flaxvision
A selection of neural network models ported from torchvision for JAX & Flax.
NobuoTsukamoto/jax_examples
Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)
google-deepmind/jraph
A Graph Neural Network Library in Jax
google-research/t5x
qubvel-org/segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.