Image segmentation models for building localization and damage assessment based on satellite imagery from the xBD dataset which was used for the xView2 challenge.

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Rank Consistent Ordinal Regression for Neural Networks with Application to Age Estimation

Time series forecasting with PyTorch

A hyperparameter optimization framework

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

AI Toolkit for Healthcare Imaging

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.