https://github.com/sseung0703/tf2-jit-compile-on-multi-gpu

Tensorflow2 training code with jit compiling on multi-GPU.
https://github.com/sseung0703/tf2-jit-compile-on-multi-gpu

jit-compilation multi-gpu-training tensorflow2 xla

Last synced: 2 days ago
JSON representation

Tensorflow2 training code with jit compiling on multi-GPU.

• Host: GitHub
• URL: https://github.com/sseung0703/tf2-jit-compile-on-multi-gpu
• Owner: sseung0703
• Created: 2020-12-23T13:15:44.000Z (almost 4 years ago)
• Default Branch: main
• Last Pushed: 2021-01-28T07:44:03.000Z (over 3 years ago)
• Last Synced: 2023-03-06T22:00:36.547Z (over 1 year ago)
• Topics: jit-compilation, multi-gpu-training, tensorflow2, xla
• Language: Python
• Homepage:
• Size: 157 KB
• Stars: 17
• Watchers: 2
• Forks: 2
• Open Issues: 0

Awesome Lists containing this project