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 ensure_on_device function #463

Merged
merged 18 commits into from
Nov 2, 2023
Merged

Remove ensure_on_device function #463

merged 18 commits into from
Nov 2, 2023

Commits on Oct 12, 2023

  1. Remove jax.device_put calls and some clean up

    @bwohlberg
    bwohlberg committed Oct 12, 2023
    Configuration menu
    Copy the full SHA
    2acf4e1 View commit details
    Browse the repository at this point in the history

  2. Remove ensure_on_device calls

    @bwohlberg
    bwohlberg committed Oct 12, 2023
    Configuration menu
    Copy the full SHA
    fa804aa View commit details
    Browse the repository at this point in the history

  3. Minor edit

    @bwohlberg
    bwohlberg committed Oct 12, 2023
    Configuration menu
    Copy the full SHA
    4fd7f73 View commit details
    Browse the repository at this point in the history

  4. Update submodule

    @bwohlberg
    bwohlberg committed Oct 12, 2023
    Configuration menu
    Copy the full SHA
    349e621 View commit details
    Browse the repository at this point in the history

Commits on Oct 27, 2023

  1. Update submodule

    @bwohlberg
    bwohlberg committed Oct 27, 2023
    Configuration menu
    Copy the full SHA
    d823be5 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into brendt/rm-ensure-on-device

    @bwohlberg
    bwohlberg committed Oct 27, 2023
    Configuration menu
    Copy the full SHA
    835f14d View commit details
    Browse the repository at this point in the history

Commits on Oct 31, 2023

  1. Remove unnecessary forced conversion to jax array

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    314ab61 View commit details
    Browse the repository at this point in the history

  2. Delete ensure_on_device function

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    66bf302 View commit details
    Browse the repository at this point in the history

  3. Delete ensure_on_device function

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    7b3ecfa View commit details
    Browse the repository at this point in the history

  4. Fix compatibility issue with older jax versions

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    1a2f2d4 View commit details
    Browse the repository at this point in the history

  5. Update submodule

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    25a270f View commit details
    Browse the repository at this point in the history

  6. Merge branch 'main' into brendt/rm-ensure-on-device

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    b25821d View commit details
    Browse the repository at this point in the history

  7. Clean up

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    b0458de View commit details
    Browse the repository at this point in the history

  8. Remove jax.device_put calls

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    ecf328c View commit details
    Browse the repository at this point in the history

  9. Remove jax.device_put calls

    @bwohlberg
    bwohlberg committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    faa28f9 View commit details
    Browse the repository at this point in the history

Commits on Nov 2, 2023

  1. Merge branch 'main' into brendt/rm-ensure-on-device

    @bwohlberg
    bwohlberg committed Nov 2, 2023
    Configuration menu
    Copy the full SHA
    2c99267 View commit details
    Browse the repository at this point in the history

  2. Address PR review comment

    @bwohlberg
    bwohlberg committed Nov 2, 2023
    Configuration menu
    Copy the full SHA
    db7b668 View commit details
    Browse the repository at this point in the history

  3. Update submodule

    @bwohlberg
    bwohlberg committed Nov 2, 2023
    Configuration menu
    Copy the full SHA
    a5fb2fd View commit details
    Browse the repository at this point in the history