← all repositories
Oneflow-Inc/oneflow

OneFlow treats massive distributed training as a tensor detail

OneFlow exists to prove that distributed training across thousands of GPUs shouldn't require rewriting code that already works on one.

9.4k stars C++ ML Frameworks
oneflow
Not currently ranked — collecting fresh signals.
star history

What it does OneFlow is a deep learning framework offering a PyTorch-like API, but built from the ground up in C++ with its own graph compiler and runtime. Its central trick is the Global Tensor, an abstraction that turns n-dimensional distributed parallelism—data, model, and pipeline splits across clusters—into a tensor property rather than a code rewrite. The project also maintains companion libraries for large transformers (Libai) and computer vision (FlowVision).

The interesting bit Instead of retrofitting distribution onto a single-device framework, OneFlow’s authors redesigned the stack from scratch so that scaling out theoretically feels like scaling up. The README is upfront that several older model zoos and benchmark suites are now outdated, which is either refreshing honesty or a warning sign, depending on your optimism.

Key highlights

  • PyTorch-compatible frontend, but execution is handled by a dedicated C++ graph compiler and runtime.
  • Global Tensor promises n-dimensional parallelism without manual sharding logic.
  • Companion libraries cover GPT, BERT, T5, and vision backbones.
  • Version 1.0.0 is out; an arXiv paper explains the architectural redesign.
  • Linux and NVIDIA-only; Windows and macOS need not apply.

Caveats

  • Multiple official model repositories are explicitly tagged “Outdated,” suggesting the ecosystem has shifted significantly between releases.
  • The README lists hardware requirements and install steps extensively, yet offers no head-to-head performance or accuracy comparisons against PyTorch, JAX, or TensorFlow.
  • Platform support is strictly Linux with CUDA arch 60+; CPU-only builds are mentioned but clearly second-class.

Verdict A plausible alternative for teams hitting the limits of PyTorch Distributed and willing to trade ecosystem maturity for a framework that treats massive parallelism as a first-class citizen. If you are not training across dozens of GPUs, you are likely not the target audience.

Frequently asked

What is Oneflow-Inc/oneflow?
OneFlow exists to prove that distributed training across thousands of GPUs shouldn't require rewriting code that already works on one.
Is oneflow open source?
Yes — Oneflow-Inc/oneflow is open source, released under the Apache-2.0 license.
What language is oneflow written in?
Oneflow-Inc/oneflow is primarily written in C++.
How popular is oneflow?
Oneflow-Inc/oneflow has 9.4k stars on GitHub.
Where can I find oneflow?
Oneflow-Inc/oneflow is on GitHub at https://github.com/Oneflow-Inc/oneflow.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.