End to End steps for adding custom ops in PyTorch.
☆24Aug 20, 2020Updated 5 years ago
Alternatives and similar repositories for pytorch_custom_op
Users that are interested in pytorch_custom_op are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- CAKE Library for constant-bandwidth matrix multiplication on CPUs☆14Apr 6, 2024Updated 2 years ago
- A intelligent matrix format designer for SpMV☆10Oct 10, 2023Updated 2 years ago
- A pseudo random number generator library written against the SYCL API.☆11Jun 11, 2019Updated 7 years ago
- Marek's approach to building AMD GPU drivers for driver development☆28Jun 1, 2026Updated 2 months ago
- A warp-oriented dynamic hash table for GPUs☆76Jan 19, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆208May 22, 2026Updated 2 months ago
- Matrix multiplication on GPUs for matrices stored on a CPU. Similar to cublasXt, but ported to both NVIDIA and AMD GPUs.☆33Apr 2, 2025Updated last year
- My tests and experiments with some popular dl frameworks.☆17Sep 11, 2025Updated 11 months ago
- Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)☆12May 26, 2020Updated 6 years ago
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- TiledLower is a Dataflow Analysis and Codegen Framework written in Rust.☆13Nov 23, 2024Updated last year
- Source code for the paper "LongGenBench: Long-context Generation Benchmark"☆24Oct 8, 2024Updated last year
- Public proposals, extensions, information and materials from the SYCL working group☆15Jan 26, 2024Updated 2 years ago
- Hardware go brrr bounded context suffix array construction algorithm☆19Nov 1, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- High-speed Bloom filters and taffy filters for C, C++, and Java☆35Aug 9, 2023Updated 3 years ago
- TiledKernel is a code generation library based on macro kernels and memory hierarchy graph data structure.☆19May 12, 2024Updated 2 years ago
- Noisy language compiler☆17Jul 31, 2024Updated 2 years ago
- Highly Efficient FFT for Exascale☆38Apr 29, 2024Updated 2 years ago
- Emulating DMA Engines on GPUs for Performance and Portability☆43May 17, 2015Updated 11 years ago
- A study for a custom convolution layer in which the x and y components of an image pixel are added to the kernel inputs.☆12Feb 21, 2020Updated 6 years ago
- Commands that will make you more comfortable with the ROCm toolkit.☆18Aug 1, 2024Updated 2 years ago
- A GPU FP32 computation method with Tensor Cores.☆27Dec 8, 2025Updated 8 months ago
- Automatic virtualization of (general) accelerators.☆47Nov 28, 2022Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Yinghan's Code Sample☆365Jul 25, 2022Updated 4 years ago
- 🐆 A compiler from AI model to RTL (Verilog) accelerator in FPGA hardware with auto design space exploration for *AdderNet*☆22May 27, 2024Updated 2 years ago
- ☆20Feb 12, 2025Updated last year
- Setting up Vscode to work with Pytorch in C/C++ with CUDA support☆25Feb 5, 2025Updated last year
- DeeperGEMM: crazy optimized version☆86May 5, 2025Updated last year
- PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)☆66Mar 24, 2025Updated last year
- ☆23Feb 18, 2025Updated last year
- CUDA实现'huawei-noah/AdderNet'的forward和backward☆17Apr 16, 2020Updated 6 years ago
- ☆20Dec 24, 2024Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Experimental OpenCL SPIR-V to OpenCL C translator☆29Mar 1, 2026Updated 5 months ago
- This is a BNN_Kernel on PyTorch for 1-bit networks in image data processing☆23Sep 28, 2019Updated 6 years ago
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆30Jan 22, 2026Updated 6 months ago
- Transformers components but in Triton☆34May 9, 2025Updated last year
- SocksDirect code repository☆20May 6, 2026Updated 3 months ago
- ngAP's artifact for ASPLOS'24☆25Jul 29, 2025Updated last year
- GitLab runner for HPC systems using ENROOT and SLURM☆33Sep 28, 2023Updated 2 years ago