Skip to main content

Data pipeline framework for Slurm-managed HPC clusters

Project description

TigerFlow

Python License: MIT CI CD

tigerflow-logo

TigerFlow is a Python framework that simplifies the creation and execution of data pipelines on Slurm-managed HPC clusters. It supports data pipelines where:

  • Each task performs embarrassingly parallel, one-to-one file processing. That is, each input file is transformed into a single output file independently of all other input files.
  • The graph of task input/output files forms an arborescence. That is, there is a single root file, and every other file depends on exactly one parent.

Designed as a continuously running service with dynamic scaling, TigerFlow minimizes the need for users to manually plan and allocate resources in advance.

Why TigerFlow?

HPC clusters are an invaluable asset for researchers who require significant computational resources. For example, computational social scientists may need to extract features (e.g., transcription embeddings) from a large volume of TikTok videos and store them in databases for downstream analysis and modeling. However, the architecture of HPC clusters can present challenges for such workflows:

  • Compute nodes often lack internet access. This prevents direct access to external APIs (e.g., LLM services provided by Google) or remote data sources (e.g., Amazon S3), requiring such tasks to be executed on a login or head node instead.

  • Compute nodes often have restricted access to file systems. Certain file systems (e.g., cold storage) may not be mounted on compute nodes. This necessitates moving or copying data to accessible locations (e.g., scratch space) before processing can occur on compute nodes.

These constraints make it difficult to design and implement end-to-end data pipelines when some steps require external API call—restricted to login/head nodes—while others depend on high-performance compute resources available only on compute nodes. TigerFlow addresses these challenges by offering a simple, unified framework for defining and running data pipelines across different types of cluster nodes.

Key Features

TigerFlow further streamlines HPC workflows by addressing common inefficiencies in traditional Slurm-based job scheduling:

  • No need to pre-batch workloads. Each Slurm task in TigerFlow runs a dynamically scalable worker cluster that automatically adapts to the incoming workload, eliminating the need for manual batch planning and tuning.
  • No need to start a new Slurm job for each file. In TigerFlow, a single Slurm job runs as a long-lived worker process that handles multiple files. It performs shared operations (e.g., setup and teardown) once, while applying file-processing logic individually to each file. This reduces idle time and resource waste from launching a separate Slurm job for every file.
  • No need to wait for all files to complete a pipeline step. In TigerFlow, files are processed individually as they arrive, supporting more flexible and dynamic workflows.

These features make TigerFlow especially well-suited for running large-scale or real-time data pipelines on HPC systems.

Quickstart

TigerFlow can be run on any HPC cluster managed by Slurm. Since it is written in Python, the system must have Python (versions 3.10 through 3.13) installed.

Installation

TigerFlow can be installed using pip:

pip install tigerflow

It can also be installed using other package managers such as uv and poetry.

Usage

Once the package is installed, tigerflow command will be available, like so:

tigerflow --help

Running the above will display an overview of the tool, including supported subcommands.

For instance, run is a subcommand for running a user-defined pipeline, and its details can be viewed by running:

tigerflow run --help

Try running the examples, starting with a simple pipeline consisting of two local tasks.

Next Steps

Check out the user guide for more details.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tigerflow-0.2.1.tar.gz (7.1 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

tigerflow-0.2.1-py3-none-any.whl (35.5 kB view details)

Uploaded Python 3

File details

Details for the file tigerflow-0.2.1.tar.gz.

File metadata

  • Download URL: tigerflow-0.2.1.tar.gz
  • Upload date:
  • Size: 7.1 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.7.15

File hashes

Hashes for tigerflow-0.2.1.tar.gz
Algorithm Hash digest
SHA256 e6fafef6159513164c879cf3bd40041d43b1e5d140551f9c6d68ffef0657f5d3
MD5 6341109ed3b42ee78cd23e2e548da195
BLAKE2b-256 6e60ebda256b1276d2b7b3be946805e0cb22f4d52eaecf19e02ecb0a9f7abf9c

See more details on using hashes here.

File details

Details for the file tigerflow-0.2.1-py3-none-any.whl.

File metadata

  • Download URL: tigerflow-0.2.1-py3-none-any.whl
  • Upload date:
  • Size: 35.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.7.15

File hashes

Hashes for tigerflow-0.2.1-py3-none-any.whl
Algorithm Hash digest
SHA256 60a12095ef6328cc4e05e6aa2b831fe3205bc0319bf73d600b0e4c13f325c46f
MD5 bd660566a1f5922a2b7cca4dba4b002b
BLAKE2b-256 8959ab0b7a8fcaf64fbca9838ec6c76dabf576bcbab12cf2ed3652041501524d

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page