M1 Apple machine support

[EdwardCuiPeacock]

Currently, TFX hard codes the tensorflow version here https://github.com/tensorflow/tfx/blob/master/tfx/dependencies.py, which specifically uses tensorflow package. However, tensorflow is not supported on M1 Apple silicon; users need to install tensorflow-macos in order to run tensorflow (does not change Python imports in scripts). Tensorflow has yet to support arm64 wheels for M1 Apple silicon (see this issue: tensorflow/tensorflow#57185). So at the meantime, it will be nice to offer certain platform-dependent flexibility that allows the M1 silicon users to use tensorflow-macos as depedency.

[singhniraj08]

@EdwardCuiPeacock,

This feature request is already in progress and team is working on priority for this, but there are internal infrastructure reasons for why this has proceeded slowly.

Rest assured, You can follow this TF forum thread for updates and we will update this thread once we have some updates.

Thank you!

[EdwardCuiPeacock]

Thank you for the response. @singhniraj08 Looking forward to being able to do fully local development on my new machine.

[yingding]

+1 , thanks for the great work. The TF 2.13 now supports apple silicon natively (https://blog.tensorflow.org/2023/07/whats-new-in-tensorflow-213-and-keras-213.html). I am looking forward to the TFX support on apple silicon soon.

[tangm]

There's been some PRs relevant to this and I've written up my experience trying to get tfx 1.14.0 working on apple silicon natively here

The relevant pull requests are:

• PR: Enable macOS M1 wheel builds google/ml-metadata#166, Issue: 143
• PR: Allow building of tfx-bsl on mac arm64 machines tfx-bsl#73, Issue: 48
• PR: Allow compilation by including arm64 macs in mac select conditions data-validation#251, Issue: 205

EDIT: wrong link, add issue links