← All projects
Training & fine-tuning

OneFlow

A deep learning framework, similar to PyTorch, for building and training AI models across many machines.

Research-backedNew this week

What it does

OneFlow is a deep learning framework: the core software used to build and train AI models. Its makers designed it to be easy to use, scalable and efficient.

Programmers write models with commands that look like PyTorch, a widely used framework. OneFlow can then spread a model across many graphics cards and machines, and speed it up with a built-in compiler.

It runs on Linux and works best with NVIDIA graphics cards, though a CPU-only version is available.

Who it's for

AI researchers and engineers who train large models across many graphics cards.

Can I use it without coding?

Developers only. It is a programming framework, used by writing Python code. New to this? Start here.

Can I run it?

Cost
FreeNothing to pay or sign up for.
Runs on
Linux
Computer
Works best on a powerful computer or one with a graphics card (GPU)
Privacy
Runs on your own device: your data stays with you
API key
Not needed

Try it

Using Claude Code, Cursor or another AI coding assistant? Paste the prompt and it will walk you through the setup.New to this?

Before you install

  • Only download from the official GitHub page. Copies with similar names exist for many popular projects.
  • License (Apache-2.0): Permissive: free to use, change and share, including for business.
  • More on staying safe

Where people are talking about it