Oneflow-Inc / one-fx
A toolkit for developers to simplify the transformation of nn.Module instances. It's now corresponding to Pytorch.fx.
☆13Updated last year
Alternatives and similar repositories for one-fx:
Users that are interested in one-fx are comparing it to the libraries listed below
- ☆11Updated last year
- OneFlow->ONNX☆42Updated last year
- ☆18Updated last year
- OneFlow Serving☆20Updated last month
- ☆13Updated last year
- TVMScript kernel for deformable attention☆24Updated 3 years ago
- ☆14Updated 10 months ago
- Datasets, Transforms and Models specific to Computer Vision☆84Updated last year
- An object detection codebase based on MegEngine.☆28Updated 2 years ago
- ☆23Updated last year
- MegEngine到其他框架的转换器☆69Updated last year
- [CVPR-2023] Towards Any Structural Pruning☆16Updated last year
- NVIDIA TensorRT Hackathon 2023复赛选题:通义千问Qwen-7B用TensorRT-LLM模型搭建及优化☆41Updated last year
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- ☆11Updated last year
- study of cutlass☆21Updated 3 months ago
- CVFusion is an open-source deep learning compiler to fuse the OpenCV operators.☆29Updated 2 years ago
- A codebase & model zoo for pretrained backbone based on MegEngine.☆33Updated last year
- An easy way to run, test, benchmark and tune OpenCL kernel files☆23Updated last year
- Training LLaMA language model with MMEngine! It supports LoRA fine-tuning!☆40Updated last year
- # Unified Normalization (ACM MM'22) By Qiming Yang, Kai Zhang, Chaoxiang Lan, Zhi Yang, Zheyang Li, Wenming Tan, Jun Xiao, and Shiliang P…☆34Updated last year
- Yet another Polyhedra Compiler for DeepLearning☆19Updated last year
- A high-throughput and memory-efficient inference and serving engine for LLMs☆16Updated 8 months ago
- ☕️ A vscode extension for netron, support *.pdmodel, *.nb, *.onnx, *.pb, *.h5, *.tflite, *.pth, *.pt, *.mnn, *.param, etc.☆12Updated last year
- ☆12Updated last year
- ☆35Updated 4 months ago
- ☆17Updated 10 months ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆53Updated 6 months ago
- 天池 NVIDIA TensorRT Hackathon 2023 —— 生成式AI模型优化赛 初赛第三名方案☆48Updated last year