Oneflow-Inc / one-fxLinks
A toolkit for developers to simplify the transformation of nn.Module instances. It's now corresponding to Pytorch.fx.
☆13Updated 2 years ago
Alternatives and similar repositories for one-fx
Users that are interested in one-fx are comparing it to the libraries listed below
Sorting:
- ☆11Updated last year
- OneFlow->ONNX☆43Updated 2 years ago
- ☆18Updated last year
- ☆13Updated 2 years ago
- OneFlow Serving☆20Updated 2 months ago
- ☆16Updated last year
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- An object detection codebase based on MegEngine.☆28Updated 2 years ago
- ☆11Updated last year
- Datasets, Transforms and Models specific to Computer Vision☆85Updated last year
- 🤗CacheDiT: A Training-free and Easy-to-use Cache Acceleration Toolbox for Diffusion Transformers🔥☆61Updated this week
- TVMScript kernel for deformable attention☆25Updated 3 years ago
- 天池 NVIDIA TensorRT Hackathon 2023 —— 生成式AI模型优化赛 初赛第三名方案☆49Updated last year
- NVIDIA TensorRT Hackathon 2023复赛选题:通义千问Qwen-7B用TensorRT-LLM模型搭建及优化☆42Updated last year
- [CVPR-2023] Towards Any Structural Pruning☆16Updated 2 years ago
- ☆23Updated 2 years ago
- study of cutlass☆21Updated 7 months ago
- ☆12Updated 2 years ago
- Awesome code, projects, books, etc. related to CUDA☆17Updated last week
- CVFusion is an open-source deep learning compiler to fuse the OpenCV operators.☆29Updated 2 years ago
- GPTQ inference TVM kernel☆40Updated last year
- Paddle Automatically Diff Precision Toolkits.☆49Updated last year
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆38Updated 2 weeks ago
- Odysseus: Playground of LLM Sequence Parallelism☆70Updated last year
- ☆36Updated 8 months ago
- A tool convert TensorRT engine/plan to a fake onnx☆39Updated 2 years ago
- ☆29Updated 4 months ago
- Training LLaMA language model with MMEngine! It supports LoRA fine-tuning!☆40Updated 2 years ago
- A CUDA kernel for NHWC GroupNorm for PyTorch☆19Updated 7 months ago
- # Unified Normalization (ACM MM'22) By Qiming Yang, Kai Zhang, Chaoxiang Lan, Zhi Yang, Zheyang Li, Wenming Tan, Jun Xiao, and Shiliang P…☆34Updated 2 years ago