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
andfn_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