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

Jump to bottom

better custom autodiff #407

Merged
merged 4 commits into from
Jul 5, 2023
Merged

better custom autodiff #407

merged 4 commits into from
Jul 5, 2023

Commits on Jul 3, 2023

  1. Now using symbolic zeros in filter_custom_{jvp,vjp}

    @patrick-kidger
    patrick-kidger committed Jul 3, 2023
    Configuration menu
    Copy the full SHA
    55eefa5 View commit details
    Browse the repository at this point in the history

Commits on Jul 4, 2023

  1. Removed unnecessary lax.select when not vmap'ing

    @patrick-kidger
    patrick-kidger committed Jul 4, 2023
    Configuration menu
    Copy the full SHA
    4a2c76b View commit details
    Browse the repository at this point in the history

  2. Improved speed of backprop through eqxi.while_loop buffers, in partic… 

    …ular avoiding JAX issue #10197
    @patrick-kidger
    patrick-kidger committed Jul 4, 2023
    Configuration menu
    Copy the full SHA
    0656c45 View commit details
    Browse the repository at this point in the history

Commits on Jul 5, 2023

  1. Fixed docs

    @patrick-kidger
    patrick-kidger committed Jul 5, 2023
    Configuration menu
    Copy the full SHA
    a200754 View commit details
    Browse the repository at this point in the history