Skip to content

1.5.0
Compare
Choose a tag to compare
@holl- holl- released this 31 Mar 16:49
· 277 commits to main since this release
1.5.0
a009b7e

The Φ-ML paper has been published! DOI

This version updates the value/variable attribute system and is compatible with TensorFlow 2.16 and the latest JAX versions.

Highlights

  • Value attributes are now used in optimization and gradient computation.
  • tensorflow_probability is no longer required for median() and quantile()
  • Added find_differences to compare trees
  • Added sort
  • grid_sample and closest_grid_values now support specifying grid dims via coordinates

Notable fixes

  • PyTorch's long-standing JIT adjoint linear solve problem has been resolved
  • Tracing JIT functions within a gradient function no longer stores the gradients (caused JAX escaped tracers)
pip install phiml==1.5.0
Assets 2
Loading