• Stars
    star
    104
  • Rank 330,604 (Top 7 %)
  • Language
    Python
  • Created about 5 years ago
  • Updated 3 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

OneFlow models for benchmarking.

OneFlow Deep Learning Benchmarks

Introduction

This repository provides OneFlow deep learning benchmark examples for CV, CTR and NLP, and more models are on the way and will be provided here when ready.

Convolutional Networks for Computer Vision Classification

Wide Deep Learning for Click-Through-Rate (CTR) Recommender Systems

BERT for Nature Language Process

GPT for Generative Pre-trained Transformer

OneFlow Benchmark Test Reports

Model DType XLA Throughput Speedup on 32 devices
ResNet50-V1.5 Float32 No 11.6k imges/sec 30.4
BERT base Pretrain Float32 No 530k tokens/sec 28.54

More Repositories

1

oneflow

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.
C++
5,888
star
2

onediff

OneDiff: A drop-in acceleration lib for ComfyUI, HF diffusers, Stable Diffusion web UI, and other diffusion models.
Python
622
star
3

libai

LiBai(ζŽη™½): A Toolbox for Large-Scale Distributed Parallel Training
Python
391
star
4

DLPerf

DeepLearning Framework Performance Profiling Toolkit
Python
275
star
5

one-yolov5

A more efficient yolov5 with oneflow backend πŸŽ‰πŸŽ‰πŸŽ‰
Python
213
star
6

models

Models and examples built with OneFlow
Python
94
star
7

vision

Datasets, Transforms and Models specific to Computer Vision
Python
82
star
8

oneflow-documentation

oneflow documentation
HTML
68
star
9

one-glm

A more efficient GLM implementation!
Python
54
star
10

oneflow_convert

OneFlow->ONNX
Python
41
star
11

oneflow-xrt

C++
22
star
12

oneflow-yolo-doc

https://start.oneflow.org/oneflow-yolo-doc
HTML
22
star
13

serving

OneFlow Serving
C++
20
star
14

oneflow-lite

C++
18
star
15

one-fx

A toolkit for developers to simplify the transformation of nn.Module instances. It's now corresponding to Pytorch.fx.
Python
13
star
16

diffusers

Python
13
star
17

oneflow_face

Python
12
star
18

conda-env

Shell
12
star
19

flow-OpCounter

Count the FLOPs & Params of your OneFlow model.
Python
11
star
20

occl

C++
11
star
21

diffusion-benchmark

Python
11
star
22

oneflow_vision_model

Python
10
star
23

one-codegeex

Python
7
star
24

Oneflow-Model-Compression

Python
7
star
25

oneflow-api-cn

Chinese Documents of OneFlow API
Python
7
star
26

oneflow-mlu

C++
7
star
27

faster-chatglm-6b

Python
6
star
28

oneflow-hip

C++
6
star
29

OneAutoTest

Auto-Test System
Shell
5
star
30

community

Stores documents used by the OneFlow developer community
5
star
31

trt_flash_attention

C++
4
star
32

oneflow_yolov3

Python
4
star
33

get-oneflow

Build or fetch pre-build outputs of OneFlow in GitHub Actions
TypeScript
3
star
34

CoModels

Python
3
star
35

text

Data loaders and abstractions for text and NLP
Python
3
star
36

oneflow_imaginaire

Implementation of NVlabs imaginaire models in Oneflow
Python
3
star
37

utensor

C++
2
star
38

comm_network

C
2
star
39

code_film

Python
1
star
40

lesson_projects

Python
1
star
41

oneflow-insiders

Repo to collect issues of OneFlow early adopters
1
star
42

oneflow-mlu-models

Python
1
star
43

manylinux-builder

TypeScript
1
star
44

Zhusuan-Oneflow

Zhusuan with backend Oneflow
Python
1
star