NN-CUDA-Example

Several simple examples for popular neural network toolkits calling custom CUDA operators.

APACHE-2.0 License

Stars
1.3K
View Code on GitHub View on X
Ecosystems: C++, Python, PyTorch, Tensorflow

Statistics for this project are still being loaded, please check back later.

Related Projects
pytorch_dart

Pytorch_Dart is a dart wrapper for Libtorch,striving to provide an experience identical to PyTorc...

15 Mar 2024 9
tch-rs

Rust bindings for the C++ api of PyTorch.

16 Feb 2019 4,227
PyTorch-Raspberry-Pi-64-OS

PyTorch installation wheels for Raspberry Pi 64 OS

26 Feb 2021 100
PyTorchTricks

Some tricks of pytorch...

23 Dec 2019 1,153
pytorch2c

A Python module for compiling PyTorch graphs to C

15 Feb 2017 86
PyTorch-build

Provide Docker build sequences of PyTorch for various environments.

18 Feb 2021 11
pytorch4raspberrypi

Cross-compilation of PyTorch armv7l (32bit) for RaspberryPi OS

29 Aug 2021 20
Open3D-build

Provide Docker build sequences of Open3D for various environments.

15 Feb 2021 14
docker_image_with_cuda10_cudnn7

Dockerfiles and manual for easy build of docker image with CUDA10.X and cuDNN7.6 to run TensorFlo...

03 May 2019 17
jetson-tensorflow-pytorch-build

Provides an environment for compiling TensorFlow or PyTorch with CUDA for aarch64 on an x86 machi...

22 Feb 2021 13
nvidia-gpu-ml-library-test

Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being p...

04 Jan 2021 15
pytorch-cpp

Just messing around with PyTorch 1.0's JIT compiler and their new C++ API Libtorch.

09 Nov 2018 19
TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

11 Mar 2020 2,359
extension-cpp

C++ extensions in PyTorch

03 Mar 2018 999
torchdistx

Torch Distributed Experimental

16 Feb 2022 116