|
Python |
3 |
8-bit CUDA functions for PyTorch |
Apr 12, 2023 |
|
Python |
1848 |
8-bit CUDA functions for PyTorch |
May 01, 2023 |
|
Python |
8 |
8-bit CUDA functions for PyTorch |
Apr 08, 2023 |
|
Python |
2 |
8-bit CUDA functions for PyTorch |
Oct 13, 2023 |
|
Python |
3 |
8-bit CUDA functions for PyTorch |
Jun 20, 2023 |
|
Python |
13 |
8-bit CUDA functions for PyTorch in Windows 10 |
May 03, 2023 |
|
C |
2 |
8-bit CUDA functions for PyTorch, modified to build on Jetson Xavier |
May 10, 2023 |
|
C++ |
13 |
CUDA Template Functions |
May 16, 2023 |
|
Cuda |
198 |
pytorch knn [cuda version] |
Apr 24, 2023 |
|
Python |
5 |
PyTorch CUDA Extension for Sparse Blocks |
Aug 07, 2022 |
|
None |
2 |
How to write cuda kernels or c functions in pytorch, especially for former caffe users. |
Aug 03, 2021 |
|
Python |
28 |
pytorch cuda extension of grid_sample1d |
Apr 06, 2023 |
|
Cuda |
2 |
an operator written in CUDA for PyTorch |
Sep 25, 2019 |
|
Dockerfile |
4 |
PyTorch 1.10.2 Dockerfile for CUDA capability 3.0 |
Feb 01, 2023 |
|
Python |
60 |
PlenOctrees construction + rendering PyTorch CUDA extension |
Aug 25, 2022 |
|
Jupyter Notebook |
2 |
Nvidia CUDA for GPU + PyTorch (latest) in Docker |
May 18, 2022 |
|
Python |
3 |
Template for CUDA / C++ extension writing with PyTorch |
Jun 25, 2021 |
|
Cuda |
14 |
Practice to write cpp/cuda extension for pytorch |
Jul 14, 2022 |
|
Jupyter Notebook |
63 |
CUDA kernels for generalized matrix-multiplication in PyTorch |
Sep 05, 2022 |
|
Cuda |
15 |
Matrix exponential in cuda for pytorch and tensorflow |
Mar 24, 2021 |
|
Python |
2 |
Utils Functions for PyTorch |
Mar 24, 2020 |
|
Julia |
2 |
Bit twiddling convenience functions. |
Jan 26, 2022 |
|
Julia |
7 |
Julia Bit Manipulation Functions |
Nov 29, 2021 |
|
None |
2 |
Tutorial for building a custom CUDA function for Pytorch |
May 13, 2021 |
|
Python |
452 |
Tutorial for building a custom CUDA function for Pytorch |
May 03, 2023 |
|
Python |
4 |
Docker image for PyTorch based on nvidia/cuda images |
Aug 17, 2022 |
|
Jupyter Notebook |
15 |
WarpRNNT loss ported in Numba CPU/CUDA for Pytorch |
Oct 22, 2022 |
|
Dockerfile |
5 |
VS Code devcontainer for Rust, GStreamer, Pytorch and Cuda |
Jun 02, 2022 |
|
None |
61 |
cuda/pytorch/openai/fake screen setup tutorial |
Jan 26, 2023 |
|
Cuda |
3 |
PyTorch port for 3D mesh rendering accelerated with C/CUDA. |
Jun 28, 2022 |
|
Python |
5 |
Reproduce and improve ChexNet by Python Pytorch,CUDA |
Nov 05, 2021 |
|
Python |
10 |
A generalized Chamfer distance implementation in CUDA/Pytorch |
Apr 17, 2023 |
|
Python |
7 |
Implement custom operators in PyTorch with cuda/c++ |
Mar 16, 2023 |
|
None |
3 |
Prebuilt wheels for PyTorch, xformers, and other libraries for Cuda 11.8 |
Mar 26, 2023 |
|
JavaScript |
2 |
Code for A bit about JavaScript functions talk |
Jul 20, 2016 |
|
JavaScript |
2 |
Functions for doing bit manipulation of typed arrays |
Sep 04, 2022 |
|
C++ |
2 |
Vector math and other CUDA helper functions for OptiX kernels |
Jan 18, 2023 |
|
Python |
9 |
Utility functions for training PyTorch models. |
May 09, 2020 |
|
Python |
2 |
Activation functions in pytorch |
Jul 21, 2023 |
|
Jupyter Notebook |
16 |
PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization |
Oct 26, 2022 |
|
C++ |
4 |
Arduino library with static functions on bit level (a.k.a. bit hacks) |
Apr 15, 2022 |
|
Python |
10 |
Pytorch Implementation of RetinaNet with CUDA accelerate nms operation. |
Jun 30, 2022 |
|
C |
69 |
GPU PyTorch TOP in TouchDesigner with CUDA-enabled OpenCV |
May 14, 2023 |
|
Python |
2 |
Non-Reference IQA implementation in Pytorch CUDA, UCIQE, UIQM, ... |
Apr 22, 2024 |
|
Cuda |
3 |
Library for profiling the execution time of CUDA official library functions |
Feb 16, 2023 |
|
Python |
5 |
DCT (discrete cosine transform) functions for pytorch |
Apr 04, 2022 |
|
Python |
30 |
Complex tensor and complex functions for pytorch. |
Jul 24, 2022 |
|
Python |
4 |
Convenience functions for working with pytorch hooks. |
May 05, 2023 |
|
Python |
42 |
custom cuda kernel for {2, 3}d relative attention with pytorch wrapper |
Jan 28, 2023 |
|
Python |
92 |
Use Jax functions in Pytorch |
Aug 05, 2022 |