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:
- OneFlow->ONNX☆43Updated 2 years ago
- ☆18Updated last year
- ☆11Updated 3 months ago
- OneFlow Serving☆20Updated 7 months ago
- ☆23Updated 2 years ago
- ☆13Updated 2 years ago
- study of cutlass☆22Updated last year
- ☆16Updated last year
- Datasets, Transforms and Models specific to Computer Vision☆90Updated last year
- ☆12Updated 2 years ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆78Updated last year
- ☆12Updated 2 years ago
- ☆15Updated 3 years ago
- High performance inference engine for diffusion models☆94Updated 2 months ago
- A Triton JIT runtime and ffi provider in C++☆29Updated last week
- ☆139Updated last year
- oneflow documentation☆69Updated last year
- A codebase & model zoo for pretrained backbone based on MegEngine.☆33Updated 2 years ago
- Yet another Polyhedra Compiler for DeepLearning☆19Updated 2 years ago
- MegEngine到其他框架的转换器☆70Updated 2 years ago
- ☆102Updated last year
- TVMScript kernel for deformable attention☆25Updated 3 years ago
- NART = NART is not A RunTime, a deep learning inference framework.☆37Updated 2 years ago
- Models and examples built with OneFlow☆100Updated last year
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Updated 2 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆124Updated 6 months ago
- ☆38Updated last year
- [CVPR-2023] Towards Any Structural Pruning☆16Updated 2 years ago
- ☆33Updated 9 months ago
- An object detection codebase based on MegEngine.☆28Updated 2 years ago