2025 Python Packaging Survey is now live!  Take the survey now

TensorFlow acceleration for Mac GPUs.

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for dingshuhan from gravatar.com dingshuhan Avatar for jhavukainen from gravatar.com jhavukainen Avatar for kulinseth from gravatar.com kulinseth Avatar for skotapati from gravatar.com skotapati

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: MIT License. Copyright © 2020-2021 Apple Inc. All rights reserved.
  • Tags tensorflow , metal , plugin
Report project as malware

Project description

Accelerate training of machine learning models with TensorFlow right on your Mac. Install the base tensorflow (tensorflow-macos which is using TensorFlow v2.5) and the tensorflow-metal PluggableDevice to accelerate training with Metal on Mac GPUs.

You can learn more about TensorFlow PluggableDevices [here](https://github.com/tensorflow/tensorflow/releases/tag/v2.5.0).

Supported by

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