Skip to content
Permalink

Comparing changes

Choose two branches to see what’s changed or to start a new pull request. If you need to, you can also or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: google-deepmind/chex
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: v0.1.4
Choose a base ref
...
head repository: google-deepmind/chex
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: v0.1.5
Choose a head ref
  • 5 commits
  • 11 files changed
  • 2 contributors

Commits on Aug 11, 2022

  1. Add support for value assertions in jitted functions. 

    PiperOrigin-RevId: 466936096
    @hbq1
    hbq1 authored and ChexDev committed Aug 11, 2022
    Copy the full SHA
    6b219f4 View commit details

Commits on Aug 22, 2022

  1. [JAX] Avoid private implementation detail _ScalarMeta. 

    The closest public approximation to type(jnp.float32) is `type[Any]`. Nothing is ever actually an instance of one of these types, either (they build `DeviceArray`s if instantiated.)

PiperOrigin-RevId: 469132170
    @hawkinsp
    hawkinsp authored and ChexDev committed Aug 22, 2022
    Copy the full SHA
    34a4494 View commit details

Commits on Aug 23, 2022

  1. [JAX] Avoid implicit references to jax._src. 

    An upcoming change to JAX means that it will no longer export jax._src by default. It still works (for now) to import modules from jax._src and refer to those, but they will not be present in JAX's namespace except via an explicit import.

Note that any use of jax._src is a use of a JAX-private API. Please use public APIs instead. This change mostly does not yet switch users onto public APIs.

PiperOrigin-RevId: 469403086
    @hawkinsp
    hawkinsp authored and ChexDev committed Aug 23, 2022
    Copy the full SHA
    e432187 View commit details

Commits on Sep 13, 2022

  1. Enable value (aka runtime/jittable) assertions in Chex. 

    Value assertions access concrete tensor values which are not available during JAX tracing. In addition, such assertions require a separate error-handling mechanism for lifting from a jitted code.

This PR introduces a wrapper (based on [checkify](https://jax.readthedocs.io/en/latest/_autosummary/jax.experimental.checkify.check.html)) that solves both problems. See [documentation](https://chex.readthedocs.io/en/latest/api.html#chex.chexify) for more detail.

It is a backward-compatible change.

PiperOrigin-RevId: 473955237
    @hbq1
    hbq1 authored and ChexDev committed Sep 13, 2022
    Copy the full SHA
    680eda5 View commit details

  2. Release v0.1.15 

    PiperOrigin-RevId: 473969358
    @hbq1
    hbq1 authored and ChexDev committed Sep 13, 2022
    Copy the full SHA
    2fd2ca3 View commit details
Loading