zenTF : A TensorFlow extension for AMD EPYC CPUs.
Project description
EARLY ACCESS: The ZenDNN TensorFlow* Plugin (zenTF) extends TensorFlow* with an innovative upgrade that's set to revolutionize performance on AMD hardware.
As of version 4.2, AMD is unveiling a game-changing upgrade to ZenDNN, introducing a cutting-edge plug-in mechanism and an enhanced architecture under the hood. This isn't just about extensions; ZenDNN's aggressive AMD-specific optimizations operate at every level. It delves into comprehensive graph optimizations, including pattern identification, graph reordering, and seeking opportunities for graph fusions. At the operator level, ZenDNN boasts enhancements with microkernels, mempool optimizations, and efficient multi-threading on the large number of AMD EPYC cores. Microkernel optimizations further exploit all possible low-level math libraries, including AOCL BLIS.
The result? Enhanced performance with respect to baseline TensorFlow*. The ZenDNN TensorFlow* Plugin is compatible with TensorFlow versions 2.16 and later.
Support
Please note that zenTF is currently in “Early Access” mode. We welcome feedback, suggestions, and bug reports. Should you have any of these, please contact us on zendnn.maintainers@amd.com
License
AMD copyrighted code in ZenDNN is subject to the Apache-2.0, MIT, or BSD-3-Clause licenses; consult the source code file headers for the applicable license. Third party copyrighted code in ZenDNN is subject to the licenses set forth in the source code file headers of such code.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
Hashes for zentf-4.2.0-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7d2370d7d7903d406ffaf0a7a286810f7536f6bd8782dfbea5f153b6422a8a11 |
|
MD5 | 5f3ecd12bc77699aa6cb2042c2ae08a7 |
|
BLAKE2b-256 | 4098fb6d590bca153a358af25b27f7ec91a528457ee6674aa2e9881421429476 |
Hashes for zentf-4.2.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | eaa77c6651e39865ec20dafd0379e813b5777c14ec14f82d18688d66e432ef13 |
|
MD5 | 8a3e4ee71cb81b835c8992c9da9747a1 |
|
BLAKE2b-256 | b5c8630719b816ecbd9618b68b37fa407928d639c8c5a3665bacb5c635cd147c |
Hashes for zentf-4.2.0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8051152a8de399926e43551adafd55b2f4e270d05f98f25f5aee4ef7706e891c |
|
MD5 | 24dedd0d19d01e433d29121f4460754d |
|
BLAKE2b-256 | ec9914ce017490f066e34e06edc0a976b8a02c99af4a0be2da9f4b6de756abe1 |
Hashes for zentf-4.2.0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 96a69104eca92a2d167cb23409712f40c52af1f7fd4d93af1147c6aa4027cab9 |
|
MD5 | b1cd40b56527b60b3b780e2439cafa10 |
|
BLAKE2b-256 | e6c5af50e9e20c28f76d102b335ccc2fe37901dacd744685db36e56b96171609 |