Skip to content

JAX IPU v0.3.16 beta3 - Poplar SDK 3.1/3.2/3.3

Latest
Latest
Compare
Choose a tag to compare
@balancap balancap released this 11 Sep 14:50
· 4 commits to jax-v0.3.16-ipu since this release
jax-v0.3.16-ipu-beta3-sdk3
8fedd0b
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

JAX on Graphcore IPU, version 0.3.16 beta3.

🔴 ⚠️ JAX on IPUs is an experimental package, not part of the official Graphcore Poplar SDK. ⚠️ 🔴

Release notes:

Improvements & bugfixes:

  • jax.debug.callback supported on IPUs (making debugging easier!) #18;
  • Fix IPU donated SRAM buffers re-use bug, leading to wrong results (instead of raising an exception) #21;
  • Improve IPU PjRt client non-standard XLA layout handling #22;
  • Add Popvision System Analyser integration;
  • Improve performance of JAX IPU asynchronous dispatch;

Requirements: Ubuntu 20.04, Graphcore Poplar SDK 3.1/3.2/3.3 and Python 3.8

TensorFlow XLA commit hash using for jaxlib compilation: xxx

Assets 6
Loading