Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.

Development repository for the Triton language and compiler

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

Code release for ConvNeXt model

Instance Segmentation by Jointly Optimizing Spatial Embeddings and Clustering Bandwidth

This repository provides an implementation of Vec2Instance over the SpaceNet challenge AOI 2 (Vegas) building footprint dataset. Vec2Instance provides a framework for parametrization of instances, …

Differentiable elastic deformations for N-dimensional images (Python, SciPy, NumPy, TensorFlow, PyTorch).