Skip to content

JAX IPU v0.3.16 beta1 - SDK 3.1
Compare
Choose a tag to compare
@balancap balancap released this 22 Feb 10:57
· 23 commits to jax-v0.3.16-ipu since this release
jax-v0.3.16-ipu-beta1-sdk31
0e89ab2
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 beta1.

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

Release notes:

  • Single IPU support;
  • Covering most JAX LAX operators;
  • Support JAX buffer donation;
  • MNIST examples;

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

Assets 4
Loading