Pytorch Custom CUDA kernel for searchsorted
☆137Oct 25, 2023Updated 2 years ago
Alternatives and similar repositories for torchsearchsorted
Users that are interested in torchsearchsorted are comparing it to the libraries listed below
Sorting:
- Efficient reservoir sampling implementation for PyTorch☆107Sep 28, 2021Updated 4 years ago
- 1D interpolation for pytorch☆176May 25, 2024Updated last year
- PyProf2: PyTorch Profiling tool☆82Jun 25, 2020Updated 5 years ago
- Interpolation routines in Pytorch.☆112Nov 30, 2020Updated 5 years ago
- ☆12Oct 8, 2020Updated 5 years ago
- (ACL-IJCNLP 2021) Convolutions and Self-Attention: Re-interpreting Relative Positions in Pre-trained Language Models.☆21Jul 13, 2022Updated 3 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Jul 3, 2019Updated 6 years ago
- Quiz code of debugging a badly-implemented neural network☆22Dec 19, 2018Updated 7 years ago
- Code for the paper: Unified Gradient Reweighting for Model Biasing with Applications to Source Separation☆14Nov 16, 2020Updated 5 years ago
- Streaming source separation for music and speech files, using the Open-Unmix LSTM architecture.☆21Dec 8, 2022Updated 3 years ago
- ☆153May 25, 2020Updated 5 years ago
- Layout-Guided Novel View Synthesis from a Single Indoor Panorama (CVPR 2021)☆36Oct 19, 2021Updated 4 years ago
- PyTorch layer-by-layer model profiler☆606May 23, 2021Updated 4 years ago
- A PyTorch re-implementation of Neural Radiance Fields☆908Jun 18, 2021Updated 4 years ago
- A GPipe implementation in PyTorch☆863Jul 25, 2024Updated last year
- Use the Force Luke! Learning to Predict Physical Forces by Simulating Effects [CVPR2020] (https://arxiv.org/pdf/2003.12045.pdf)☆74Oct 3, 2023Updated 2 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,123Apr 20, 2022Updated 3 years ago
- ☆164Dec 13, 2023Updated 2 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago
- Python bindings for minimp3☆16Sep 11, 2023Updated 2 years ago
- Towers of Babel: Combining Images, Language, and 3D Geometry for Learning Multimodal Vision. ICCV 2021.☆43Apr 30, 2024Updated last year
- TVMScript kernel for deformable attention☆25Dec 15, 2021Updated 4 years ago
- Hierarchical configuration framework for Python☆21Sep 13, 2021Updated 4 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆75Mar 7, 2019Updated 6 years ago
- This repository contains the code for the CVPR 2020 paper "Differentiable Volumetric Rendering: Learning Implicit 3D Representations wit…☆824Nov 9, 2021Updated 4 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆331Jun 13, 2022Updated 3 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆255May 10, 2023Updated 2 years ago
- Tutorial for building a custom CUDA function for Pytorch☆522Jan 25, 2019Updated 7 years ago
- Manifold-Mixup implementation for fastai V2☆17Oct 1, 2020Updated 5 years ago
- [ECCV'18] 3DMV: Joint 3D-Multi-View Prediction for 3D Semantic Scene Segmentation☆212Feb 22, 2022Updated 4 years ago
- ☆51Nov 3, 2024Updated last year
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,159Feb 6, 2026Updated last month
- Official repo for AutoInt: Automatic Integration for Fast Neural Volume Rendering in CVPR 2021☆189Nov 25, 2022Updated 3 years ago
- Adding positional encoding to the input preserves sharp edges in the image☆208Jan 4, 2022Updated 4 years ago
- Code for the Thermodynamic Variational Objective☆26Jun 21, 2022Updated 3 years ago
- Open source repository for the code accompanying the paper 'Non-Rigid Neural Radiance Fields Reconstruction and Novel View Synthesis of a…☆365May 17, 2022Updated 3 years ago
- PyTorch Extension Library of Optimized Scatter Operations☆1,727Jan 21, 2026Updated last month
- logboard: Monitor and Compare Logs on Browser/Terminal.☆21Sep 19, 2019Updated 6 years ago