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

Remove uses of deprecated jax.lax.tie_in #1796

Merged
merged 1 commit into from
Feb 1, 2024
Merged

Remove uses of deprecated jax.lax.tie_in #1796

merged 1 commit into from
Feb 1, 2024

Commits on Feb 1, 2024

  1. Remove uses of deprecated jax.lax.tie_in 

    This has been a no-op since jax v0.2.0, and passes the second argument through unchanged. `tie_in` will be deprecated as of jax v0.4.24; see jax-ml/jax#19413

PiperOrigin-RevId: 603220025
    @copybara-github
    Jake VanderPlas authored and copybara-github committed Feb 1, 2024
    Configuration menu
    Copy the full SHA
    a6a508e View commit details
    Browse the repository at this point in the history