Jittor / cuttLinks
CUDA Tensor Transpose (cuTT) library
☆10Updated 4 years ago
Alternatives and similar repositories for cutt
Users that are interested in cutt are comparing it to the libraries listed below
Sorting:
- ☆10Updated 5 years ago
- ☆16Updated 5 years ago
- A communication library for deep learning☆51Updated last year
- Jittor object detection ssd(Single Shot MultiBox Detector) implemention.☆14Updated 4 years ago
- [ICLR 2021] "UMEC: Unified Model and Embedding Compression for Efficient Recommendation Systems" by Jiayi Shen, Haotao Wang*, Shupeng Gui…☆39Updated 4 years ago
- The official implementation of You Only Compress Once: Towards Effective and Elastic BERT Compression via Exploit-Explore Stochastic Natu…☆48Updated 4 years ago
- Code of "Visualizing and Understanding Object Detecor"☆20Updated 4 years ago
- ☆19Updated 4 years ago
- This repository contains the code for the paper in Findings of EMNLP 2021: "EfficientBERT: Progressively Searching Multilayer Perceptron …☆33Updated 2 years ago
- ☆11Updated 5 years ago
- A simple middleware to improving GPU utilization then speedup online inference.☆19Updated 4 years ago
- ☆41Updated 4 years ago
- argparse extension for hpman☆17Updated 3 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated last year
- profile tools for pytorch nn models☆42Updated 5 years ago
- ☆14Updated 3 years ago
- Slides with modifications for a course at Tsinghua University.☆62Updated 3 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 4 years ago
- Implementation of Neural Arithmetic Logic Units (https://arxiv.org/pdf/1808.00508.pdf)☆31Updated 7 years ago
- An object detection codebase based on MegEngine.☆28Updated 3 years ago
- GPU implementation of Xnor network on inference level.☆22Updated 5 years ago
- PyTorch implementation of MLP-Mixer☆37Updated 4 years ago
- Official Pytorch Implementation for the paper 'SUPER-ADAM: Faster and Universal Framework of Adaptive Gradients'☆17Updated 4 years ago
- Inference framework for MoE layers based on TensorRT with Python binding☆41Updated 4 years ago
- Colab notebooks for d2l-book☆11Updated 6 years ago
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆120Updated 4 years ago
- MixPath: A Unified Approach for One-shot Neural Architecture Search☆30Updated 5 years ago
- A codebase & model zoo for pretrained backbone based on MegEngine.☆32Updated 2 years ago
- WeightNet: Revisiting the Design Space of Weight Networks☆19Updated 4 years ago
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 6 years ago