Stars
👋 Puncc is a python library for predictive uncertainty quantification using conformal prediction.
🔥 LeetCode for PyTorch — practice implementing softmax, attention, GPT-2 and more from scratch with instant auto-grading. Jupyter-based, self-hosted or try online.
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
PyTorch Implementation of Diffusion Schrodinger Bridge Matching
Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services
a generalist algorithm for cellular segmentation with human-in-the-loop capabilities
TorchCFM: a Conditional Flow Matching library
Pytorch implementation of Hyperspherical Variational Auto-Encoders
A Python toolkit for subcellular analysis of spatial transcriptomics data
This repository contains PyTorch implementations of various random feature maps for dot product kernels.
Learning Deep Representations of Data Distributions
scPortrait is a scalable toolkit to generate single-cell representations from raw microscopy images
Transformer Based Embeddings of Wasserstein Distance
ruptures: change point detection in Python
cell type specific, trait-associated gene regulation
Hackable and optimized Transformers building blocks, supporting a composable construction.
Pytorch library for fast transformer implementations
An implementation of Performer, a linear attention-based transformer, in Pytorch
rapids-singlecell: GPU-accelerated framework for scRNA analysis
PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
Simple, safe way to store and distribute tensors
PyTorch Extension Library of Optimized Scatter Operations
Probabilistic cell segmentation for in situ spatial transcriptomics
Official implementation of the paper: "DeFoG: Discrete Flow Matching for Graph Generation"