TensorFlow and oneDNN in Partnership
Rapid growth in AI and machine learning innovations and workloads necessitates constant developments in both software and hardware infrastructure. TensorFlow, Google’s end-to-end open-source machine learning framework, and oneDNN have been collaborating closely to ensure users can fully utilize new hardware features and accelerators, with a focus on x86 architectures. This talk will cover recent projects such as int8 (AVX512_VNNI) and bfloat16 (AVX512_BF16) vectorization support, bringing custom oneDNN operations to vanilla TensorFlow, and the upcoming Intel XPU device plug-in for TensorFlow.

Penporn Koanantakool
Google