/tfx

TFX is an end-to-end platform for deploying production ML pipelines

Primary LanguagePythonApache License 2.0Apache-2.0

TFX

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

Documentation

User Documentation

Please see the TFX User Guide.

Development References

Roadmap

The TFX Roadmap, which is updated quarterly.

Release Details

For detailed previous and upcoming changes, please check here

Requests For Comment

For designs, we started to publish RFCs under the Tensorflow community.

Examples

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 tensorflow-data-validation tensorflow-model-analysis tensorflow-metadata tensorflow-transform ml-metadata apache-beam[gcp] pyarrow tfx-bsl
GitHub master nightly (1.x / 2.x) 0.15.0 0.15.2 0.15.0 0.15.0 0.15.0 2.16.0 0.14.0 0.15.1
0.15.0 1.15.0 / 2.0.0 0.15.0 0.15.2 0.15.0 0.15.0 0.15.0 2.16.0 0.14.0 0.15.1
0.14.0 1.14.0 0.14.1 0.14.0 0.14.0 0.14.0 0.14.0 2.14.0 0.14.0 n/a
0.13.0 1.13.1 0.13.1 0.13.2 0.13.0 0.13.0 0.13.2 2.12.0 n/a n/a
0.12.0 1.12 0.12.0 0.12.1 0.12.1 0.12.0 0.13.2 2.10.0 n/a n/a