Tensors and Dynamic neural networks in Python with strong GPU acceleration

A latent text-to-image diffusion model

Caffe: a fast open framework for deep learning.

⚡ A Fast, Extensible Progress Bar for Python and CLI

Latex code for making neural networks diagrams

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use th…

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

Code release for ConvNeXt model

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models l…

Publication-ready NN-architecture schematics.

AlphaFold 3 inference pipeline.

Denoising Diffusion Probabilistic Models

Statistical Rethinking Course for Jan-Mar 2023

Diffusion Models in Medical Imaging (Published in Medical Image Analysis Journal)

a generalist algorithm for cellular segmentation with human-in-the-loop capabilities

Deep and online learning with spiking neural networks in Python

An unofficial Python wrapper for OpenAI's ChatGPT API

[CVPR 2023] Official implementation of the paper "Mask DINO: Towards A Unified Transformer-based Framework for Object Detection and Segmentation"

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation.

Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch

Medical Image Segmentation with Diffusion Model

Brian is a free, open source simulator for spiking neural networks.

[ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation

DiffusionFastForward: a free course and experimental framework for diffusion-based generative models

Medical SAM 2: Segment Medical Images As Video Via Segment Anything Model 2

A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".