Skip to content

Releases: graphcore-research/jax-scalify

JAX Scalify v0.1.1

17 Jul 12:06
257760a
Compare
Choose a tag to compare

What's Changed

  • Using setuptools-scm to generate JAX Scalify version. by @balancap in #127
  • Adding Github action to publish to PyPi. by @balancap in #128

Full Changelog: v0.1...v0.1.1

JAX Scalify v0.1

17 Jul 11:21
186bf8d
Compare
Choose a tag to compare
JAX Scalify v0.1 Pre-release
Pre-release

What's Changed

  • Adding fwd/bwd cast methods compatible with FP8. by @balancap in #120
  • Fix import bug issue with jax.stages.ArgInfo in past JAX versions. by @balancap in #122
  • Adding back fn_fwd_identity_bwd and fn_bwd_identity_fwd for backw… by @balancap in #123
  • Simplify pyproject.toml setuptools config, using default configuration by @balancap in #124
  • Following official PyPa example to improve pyproject.toml by @balancap in #125
  • Final small fixes for PyPi twice upload (version 0.1). by @balancap in #126

Full Changelog: v0.0.1...v0.1

JAX Scalify v0.0.1

28 Jun 13:28
c327a2f
Compare
Choose a tag to compare
JAX Scalify v0.0.1 Pre-release
Pre-release

JAX Scalify alpha release 0.0.1, compatible with Graphcore IPU Mk2.