Context Manager to profile the forward and backward times of PyTorch's nn.Module
☆83Oct 10, 2023Updated 2 years ago
Alternatives and similar repositories for torchnnprofiler
Users that are interested in torchnnprofiler are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆10Dec 16, 2022Updated 3 years ago
- 📥 🎯 (1,4/4) an MLIR-based toolchain with Vitis HLS LLVM input/output targeting FPGAs.☆15Nov 15, 2022Updated 3 years ago
- Binary Neural Network-based COVID-19 Face-Mask Wear and Positioning Predictor on Edge Devices☆12Jul 1, 2021Updated 5 years ago
- Benchmark PyTorch Custom Operators☆14Jul 6, 2023Updated 3 years ago
- Code of "What Images are More Memorable to Machines?"☆15Feb 13, 2023Updated 3 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Code for paper "FuSeConv Fully Separable Convolutions for Fast Inference on Systolic Arrays" published at DATE 2021☆18Aug 23, 2021Updated 4 years ago
- ☆25Apr 3, 2023Updated 3 years ago
- Factorized Neural Layers☆31Jul 11, 2023Updated 3 years ago
- SOTA Learning-augmented Systems☆37May 21, 2022Updated 4 years ago
- Spatial Spectral Machine Learning☆14Oct 15, 2025Updated 9 months ago
- A Python-like programming language for testing and experimenting with concurrent programs.☆34Apr 7, 2026Updated 3 months ago
- PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo☆17Mar 13, 2023Updated 3 years ago
- Official Pytorch implementation of the paper: "Locally Shifted Attention With Early Global Integration"☆15Dec 20, 2021Updated 4 years ago
- Computer Architecture -VLSI -Verilog Codes-Xilinx-Irsim☆14May 8, 2021Updated 5 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Official code for the paper "Does CLIP's Generalization Performance Mainly Stem from High Train-Test Similarity?" (ICLR 2024)☆11Aug 26, 2024Updated last year
- A simple framework for distributed reinforcement learning in PyTorch.☆16Apr 24, 2020Updated 6 years ago
- ☆24Jun 30, 2023Updated 3 years ago
- ☆52May 19, 2025Updated last year
- A easy general acc.☆18Mar 22, 2021Updated 5 years ago
- Wanwu models release, code will be released soon☆24Aug 24, 2022Updated 3 years ago
- ☆17Jul 2, 2021Updated 5 years ago
- Code companion for the RL Post-Training Handbook - training reasoning models on a single GPU☆19Jan 30, 2026Updated 5 months ago
- Official source code of the INTERSPEECH 2023 paper: "Audio-Visual Speech Separation in Noisy Environments with a Lightweight Iterative Mo…☆20Sep 1, 2023Updated 2 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Presents an optimized Apache Beam pipeline for generating sentence embeddings (runnable on Cloud Dataflow).☆20Mar 7, 2022Updated 4 years ago
- ☆14Aug 29, 2023Updated 2 years ago
- A GPU performance profiling tool for PyTorch models☆22Jul 5, 2022Updated 4 years ago
- An MLIR-based compiler from C/C++ to AMD-Xilinx Versal AIE☆17Aug 5, 2022Updated 3 years ago
- ☆14Jan 25, 2026Updated 5 months ago
- DCGAN Implementation using PyTorch in both C++ and Python☆15May 22, 2020Updated 6 years ago
- An opinionated NLP research template☆10Aug 29, 2024Updated last year
- ☆24Nov 18, 2025Updated 8 months ago
- Python implementation of the random-walk inductive classification algorithm Modified Adsorption from P. Talukdar☆15Jul 30, 2014Updated 11 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.
- Research and development for optimizing transformers☆132Feb 16, 2021Updated 5 years ago
- Workshop materials created for the Data Science Club at UWaterloo.☆17Jun 22, 2022Updated 4 years ago
- ☆46May 24, 2025Updated last year
- A lightweight MLIR Python frontend with support for PyTorch☆28Sep 3, 2024Updated last year
- Terraform module to create an Elastic Kubernetes (EKS) cluster and associated worker instances on AWS☆14Aug 26, 2020Updated 5 years ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆15Aug 31, 2023Updated 2 years ago
- Artifact for USENIX ATC'23: TC-GNN: Bridging Sparse GNN Computation and Dense Tensor Cores on GPUs.☆58Oct 16, 2023Updated 2 years ago