Hpu (#292)

* HPU changes * HPU pins * Use HPU device * Gaudi Tweaks * Remove pinnning * call prepare again * Add docker image for HPU * Tweaks * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - * - --------- Co-authored-by: Your Name <[email protected]>
mila-iqia · Nov 21, 2024 · 798a186 · 798a186
1 parent 4fdf736
commit 798a186
Show file tree

Hide file tree

Showing 41 changed files with 5,584 additions and 582 deletions.
diff --git a/.pin/constraints-hpu-torch.txt b/.pin/constraints-hpu-torch.txt
diff --git a/benchmarks/brax/requirements.hpu.txt b/benchmarks/brax/requirements.hpu.txt