Oneflow-Inc / models
Models and examples built with OneFlow
☆94Updated last month
Related projects: ⓘ
- ☆70Updated 9 months ago
- LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training☆389Updated this week
- oneflow documentation☆68Updated 2 months ago
- ☆205Updated last year
- Simple Dynamic Batching Inference☆145Updated 2 years ago
- Transformer related optimization, including BERT, GPT☆58Updated 11 months ago
- OneFlow models for benchmarking.☆104Updated last month
- Datasets, Transforms and Models specific to Computer Vision☆82Updated 10 months ago
- ☆123Updated 3 months ago
- ☆140Updated 4 months ago
- Transformer related optimization, including BERT, GPT☆39Updated last year
- export llama to onnx☆91Updated 3 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆205Updated 3 months ago
- Transformer related optimization, including BERT, GPT☆17Updated last year
- Tutorials for writing high-performance GPU operators in AI frameworks.☆118Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆309Updated this week
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆56Updated last year
- ☆60Updated last month
- OneFlow->ONNX☆41Updated last year
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆20Updated last week
- FlagScale is a large model toolkit based on open-sourced projects.☆129Updated this week
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆451Updated 6 months ago
- ☆90Updated 6 months ago
- FlagGems is an operator library for large language models implemented in Triton Language.☆246Updated this week
- Running BERT without Padding☆456Updated 2 years ago
- DeepLearning Framework Performance Profiling Toolkit☆276Updated 2 years ago
- Easy Parallel Library (EPL) is a general and efficient deep learning framework for distributed model training.☆260Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆57Updated 2 months ago
- ☆22Updated last year
- ☆82Updated last year