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

Update to solve jax deprecations #1766

Merged
merged 7 commits into from
Apr 11, 2024
Merged

Update to solve jax deprecations #1766

merged 7 commits into from
Apr 11, 2024

Commits on Apr 11, 2024

  1. netket:update to jax.tree_util.tree_map

    @PhilipVinc
    PhilipVinc committed Apr 11, 2024
    Configuration menu
    Copy the full SHA
    33adaac View commit details
    Browse the repository at this point in the history

  2. update tests to jax.tree_util.tree_map

    @PhilipVinc
    PhilipVinc committed Apr 11, 2024
    Configuration menu
    Copy the full SHA
    9dff9d9 View commit details
    Browse the repository at this point in the history

  3. docs: update to jax.tree.map

    @PhilipVinc
    PhilipVinc committed Apr 11, 2024
    Configuration menu
    Copy the full SHA
    707a4ce View commit details
    Browse the repository at this point in the history

  4. Benchmarks

    @PhilipVinc
    PhilipVinc committed Apr 11, 2024
    Configuration menu
    Copy the full SHA
    60f9676 View commit details
    Browse the repository at this point in the history

  5. PRNGKey() -> key()

    @PhilipVinc
    PhilipVinc committed Apr 11, 2024
    Configuration menu
    Copy the full SHA
    83f175c View commit details
    Browse the repository at this point in the history

  6. fix issue in VMC from scratch tutorial

    @PhilipVinc
    PhilipVinc committed Apr 11, 2024
    Configuration menu
    Copy the full SHA
    6a4d8c3 View commit details
    Browse the repository at this point in the history

  7. format

    @PhilipVinc
    PhilipVinc committed Apr 11, 2024
    Configuration menu
    Copy the full SHA
    893f7e4 View commit details
    Browse the repository at this point in the history