csarofeen / pytorchLinks
Tensors and Dynamic neural networks in Python with strong GPU acceleration
☆26Updated 2 years ago
Alternatives and similar repositories for pytorch
Users that are interested in pytorch are comparing it to the libraries listed below
Sorting:
- ☆40Updated 7 months ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆346Updated this week
- Automatically insert nvtx ranges to PyTorch models☆18Updated 4 years ago
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆134Updated last year
- Python bindings for NVTX☆66Updated 2 years ago
- ☆50Updated last year
- A tensor-aware point-to-point communication primitive for machine learning☆260Updated this week
- Convert nvprof profiles into about:tracing compatible JSON files☆70Updated 4 years ago
- oneCCL Bindings for Pytorch*☆99Updated this week
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆180Updated 3 weeks ago
- MLIR-based partitioning system☆115Updated this week
- A library of GPU kernels for sparse matrix operations.☆270Updated 4 years ago
- An extension library of WMMA API (Tensor Core API)☆99Updated last year
- ☆171Updated last year
- Assembler for NVIDIA Volta and Turing GPUs☆226Updated 3 years ago
- PyTorch RFCs (experimental)☆134Updated 2 months ago
- System for automated integration of deep learning backends.☆47Updated 2 years ago
- ☆144Updated 6 months ago
- Ahead of Time (AOT) Triton Math Library☆75Updated this week
- PyTorch C++ API Documentation☆232Updated this week
- ☆62Updated 7 months ago
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆85Updated last year
- PyTorch interface for the IPU☆180Updated last year
- Experimental projects related to TensorRT☆109Updated this week
- Benchmark scripts for TVM☆75Updated 3 years ago
- ☆420Updated this week
- Customized matrix multiplication kernels☆56Updated 3 years ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆434Updated this week
- ☆69Updated 2 years ago