PyPI page
Home page
Author:
Vrushank Desai
License:
Summary:
A library to showcase the use of custom CUDA extensions & CUDA graphs to accelerate the inference of Diffusion Policy.
Latest version:
1.5
Required dependencies:
diffusers
|
gdown
|
gym
|
ipython
|
numpy
|
opencv-python
|
pygame
|
pymunk
|
scikit-image
|
scikit-video
|
shapely
|
torch
|
torchvision
|
tqdm
|
zarr
Downloads last day:
12
Downloads last week:
53
Downloads last month:
85