graphcore-research / jax-experimental

JAX for Graphcore IPU (experimental)
https://github.com/graphcore-research/jax-experimental#readme
Apache License 2.0
21 stars 2 forks source link