• Stars
    star
    104
  • Rank 319,043 (Top 7 %)
  • Language
    Python
  • Created over 4 years ago
  • Updated over 1 year 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,647
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
376
star
4

DLPerf

DeepLearning Framework Performance Profiling Toolkit
Python
267
star
5

one-yolov5

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

models

Models and examples built with OneFlow
Python
92
star
7

vision

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

oneflow-documentation

oneflow documentation
HTML
67
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

serving

OneFlow Serving
C++
19
star
13

oneflow-yolo-doc

https://start.oneflow.org/oneflow-yolo-doc
HTML
18
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

oneflow_face

Python
12
star
17

conda-env

Shell
12
star
18

flow-OpCounter

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

oneflow_vision_model

Python
10
star
20

diffusion-benchmark

Python
10
star
21

occl

C++
8
star
22

one-codegeex

Python
7
star
23

faster-chatglm-6b

Python
6
star
24

oneflow-api-cn

Chinese Documents of OneFlow API
Python
6
star
25

oneflow-mlu

C++
6
star
26

oneflow-hip

C++
6
star
27

OneAutoTest

Auto-Test System
Shell
5
star
28

Oneflow-Model-Compression

Python
5
star
29

community

Stores documents used by the OneFlow developer community
5
star
30

oneflow_yolov3

Python
4
star
31

diffusers

Python
4
star
32

trt_flash_attention

C++
3
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