Home/Model Training/mesh

mesh

tensorflow/mesh

Mesh TensorFlow: Model Parallelism Made Easier

GitHub

GraphCanon updated 2w · GitHub synced 2w

1.6k stars255 forksLast push 2y Python Apache-2.0

Decision brief

Mesh TensorFlow supports simplified model parallelism across multiple devices in Python under the Apache-2.0 license.

Good fit when

When working on large models that benefit from being split across many devices.
For developers already using TensorFlow who need to easily manage model parallelism.

Avoid when

If you are looking for a tool that simplifies other aspects of machine learning beyond model-parallel computation.
For projects with limited GPU/TPU resources where multi-device parallelism is not required.

Observed Jul 12, 2026 · Source: enrich:decision_facts

mesh alternatives Trust report

Verify the decision

Maintenance and security

Full trust report

Maintenance: Archived (993d since push); As of 2w
Provenance: Not a fork · Organization account; As of 2w
Security (OSV): No lockfile; As of 1mo

Public GitHub metadata and optional OSV scans. Signals, not a guarantee. Trust methodology.

Install

pip install mesh

PyPI

How it fits your stack(1)

Typed graph edges - alternatives, integrations, successors, and dependencies. Ranked by relationship type, not raw GitHub stars.

Integrates

tensorflow

Mesh TensorFlow is a specialized library designed to work with TensorFlow and simplify model parallelism, allowing for efficient distributed training of very large models.

integrates

Relationship graph

Optional deeper exploration of typed edges and category neighbours.

tensorflow197k★

pytorch-lightning31k★

IntegratesRelated

pytorch-lightning31k★

IntegratesRelated

Similar tools

Same-category neighbours not already linked as typed edges.

pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

31k★

Megatron-LM

Ongoing research training transformer models at scale

17k★

accelerate

A tool for launching, training, and using PyTorch models with ease on various devices, configurations, including mixed precision support.

9.8k★

Evidence and technical details

Sourced facts, taxonomy, compatibility claims, README excerpt, and machine-readable endpoints.

Overview

A library from the TensorFlow ecosystem designed for simplifying model parallelism across multiple devices.

Capability facts

Languages: python
Source: github.language · Aug 7, 2026

Compatibility

Sourced claims from the README excerpt - not unsourced marketing copy.

Python runtimePython

Source: README excerpt (regex_v1, Aug 7, 2026)

pip install mesh-tensorflow

Source link

README

Installation

To install the latest stable version, run

pip install mesh-tensorflow

To install the latest development version, run

pip install -e "git+https://github.com/tensorflow/mesh.git#egg=mesh-tensorflow"

Installing mesh-tensorflow does not automatically install or update TensorFlow. We recommend installing it via pip install tensorflow or pip install tensorflow-gpu. See TensorFlow’s installation instructions for details. If you're using a development version of Mesh TensorFlow, you may need to use TensorFlow's nightly package (tf-nightly).

For agents

This page has a .md twin and JSON over the API.

/tools/tensorflow-mesh.md GET /api/graphcanon/tools/tensorflow-mesh GET /api/graphcanon/graph?tool=tensorflow-mesh