TorchEx icon indicating copy to clipboard operation
TorchEx copied to clipboard

verified publisher icon Abyssaledge
→

Metadata

Collection of PyTorch customized operators.

TorchEx

Collection of PyTorch customized operators for personal projects.

Supported GPU Operators:

  • Weighted NMS used in RangeDet.
  • IngroupIndices used in SST.
  • DynamicPointPooling (Group Correction) used in FSD.
  • Connected Component Labeling used in FSD.

Usage

With PyTorch installation, simply run pip install -e ..

← Metadata

33
Stars
2
Forks
Watchers

Owner

verified publisher icon Abyssaledge

Metadata

Collection of PyTorch customized operators.

Back