Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cache tracing of (sub)calls when forming a jaxpr #9181

Closed
wants to merge 3 commits into from
Closed

cache tracing of (sub)calls when forming a jaxpr #9181

wants to merge 3 commits into from

Commits on Jan 13, 2022

  1. add axis env state to cache keys, fixes jax-ml#9187

    @mattjj
    mattjj committed Jan 13, 2022
    Configuration menu
    Copy the full SHA
    240079f View commit details
    Browse the repository at this point in the history

  2. add axis env state to cache keys, fixes jax-ml#9187

    @mattjj
    mattjj committed Jan 13, 2022
    Configuration menu
    Copy the full SHA
    2d4e797 View commit details
    Browse the repository at this point in the history

  3. cache tracing of (sub)calls when forming a jaxpr

    @mattjj
    mattjj committed Jan 13, 2022
    Configuration menu
    Copy the full SHA
    8b5a9f5 View commit details
    Browse the repository at this point in the history