Skip to main content

TensorFlow Extended (TFX) is a TensorFlow-based general-purpose machine learning platform implemented at Google

Project description


Python PyPI

TensorFlow Extended (TFX) is a Google-production-scale machine learning platform based on TensorFlow. It provides a configuration framework to express ML pipelines consisting of TFX components. TFX pipelines can be orchestrated using Apache Airflow and Kubeflow Pipelines. Both the components themselves as well as the integrations with orchestration systems can be extended.

TFX components interact with a ML Metadata backend that keeps a record of component runs, input and output artifacts, and runtime configuration. This metadata backend enables advanced functionality like experiment tracking or warmstarting/resuming ML models from previous runs.

TFX Components


Please see the TFX User Guide.


Compatible versions

The following table describes how the tfx package versions are compatible with its major dependency PyPI packages. This is determined by our testing framework, but other untested combinations may also work.

tfx tensorflow apache-beam[gcp] tensorflow-data-validation tensorflow-model-analysis tensorflow-transform ml-metadata tensorflow-metadata
GitHub master nightly (1.x) 2.10.0 0.12.0 0.12.1 0.12.0 0.13.2 0.12.1
0.12.0 1.12.0 2.10.0 0.12.0 0.12.1 0.12.0 0.13.2 0.12.1

Download files

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

Filename, size & hash SHA256 hash help File type Python version Upload date
tfx-0.12.0-py2-none-any.whl (195.2 kB) Copy SHA256 hash SHA256 Wheel py2

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page