Skip to content

princeton-ddss/tigerflow-ml

Repository files navigation

tigerflow-ml

CI PyPI Docs

ML tasks for TigerFlow — private cloud ML APIs on HPC infrastructure.

Installation

pip install tigerflow-ml

Tasks

Task Description Entry Point
OCR Extract text from images and PDFs ocr / ocr-local
Translation Translate text documents translate / translate-local
Transcription Transcribe audio to text transcribe / transcribe-local
Object Detection Detect objects in images and videos detect / detect-local

Each task provides both a Slurm variant (for HPC) and a Local variant (for development).

Usage

After installation, tasks are automatically discoverable via:

tigerflow tasks list

Run a task directly:

python -m tigerflow_ml.text.ocr.slurm --help
python -m tigerflow_ml.text.translate.slurm --help
python -m tigerflow_ml.audio.transcribe.slurm --help
python -m tigerflow_ml.image.detect.slurm --help

Development

uv sync --group dev
uv run pre-commit run --all-files
uv run pytest tests

About

TigerFlow task library extension for ML

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 2

  •  
  •  

Languages