torch-queue v0.1
A protoype batched differentiable priority queue in raw pytorch, and a simple djikstra implementation.
Roughly based on BGPQ: A Heap-Based Priority Queue Design for GPUs
You might also like...
Official Pytorch implementation of ICLR 22 paper "Fast Differentiable Matrix Square Root"
Official Pytorch implementation of ICLR 22 paper "Fast Differentiable Matrix Square Root"
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning
TorchRL Disclaimer This library is not officially released yet and is subject to change. The features are available before an official release so that users and collaborators can get early access and provide
A PyTorch Implementation of i-MAE: Linearly Separable Representation in MAE
A PyTorch Implementation of i-MAE: Linearly Separable Representation in MAE
Official ECCV 2022 repository for SUPR: A Sparse Unified Part-Based Human Representation
SUPR: A Sparse Unified Part-Based Human Representation (ECCV 2022) TLDR: We release a full suite of state of the art models ( 18 models ): A body model, hand model, head model and a foot
Yet another PyTorch implementation of Stable Diffusion
stable-diffusion-pytorch Yet another PyTorch implementation of Stable Diffusion. I tried my best to make the codebase minimal, self-contained, consistent, hackable, and easy to read. Features are pruned if not needed in Stable Diffusion
A parallel ODE solver for PyTorch
A Parallel ODE Solver for PyTorch torchode is a suite of single-step ODE solvers such as dopri5 or tsit5 that are compatible with PyTorch’s JIT compiler and parallelized across a batch.
PyTorch implementation and pre-trained models for ASP - Autoregressive Structured Prediction with Language Models
PyTorch implementation and pre-trained models for ASP - Autoregressive Structured Prediction with Language Models
Reliable Decision from Multiple Subtasks through Threshold Optimization: Content Moderation in the Wild
TruSThresh Official PyTorch implementation of “Reliable Deicision from Multiple Subtasks through Threshold Optimization: Content Moderation in the Wild“ (WSDM’23). Installation pip install -e ./ Quick Example import numpy as np