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

Add default arg values to JAX decorator #5115

Merged
merged 21 commits into from
Oct 27, 2023
Merged

Add default arg values to JAX decorator #5115

merged 21 commits into from
Oct 27, 2023

Commits on Oct 11, 2023

  1. Add JAX getting started tutorial 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 11, 2023
    Configuration menu
    Copy the full SHA
    55bb46c View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'nvidia/main' into add_jax_getting_started

    @awolant
    awolant committed Oct 11, 2023
    Configuration menu
    Copy the full SHA
    e11f74e View commit details
    Browse the repository at this point in the history

  3. Merge remote-tracking branch 'nvidia/main' into add_jax_getting_started

    @awolant
    awolant committed Oct 11, 2023
    Configuration menu
    Copy the full SHA
    5f48fec View commit details
    Browse the repository at this point in the history

Commits on Oct 12, 2023

  1. Add JAX Getting Started tutorial 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 12, 2023
    Configuration menu
    Copy the full SHA
    be69067 View commit details
    Browse the repository at this point in the history

Commits on Oct 17, 2023

  1. Fix review comments 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 17, 2023
    Configuration menu
    Copy the full SHA
    578e0ac View commit details
    Browse the repository at this point in the history

  2. Fix 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 17, 2023
    Configuration menu
    Copy the full SHA
    460b9dc View commit details
    Browse the repository at this point in the history

  3. Fix 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 17, 2023
    Configuration menu
    Copy the full SHA
    a3905b7 View commit details
    Browse the repository at this point in the history

Commits on Oct 18, 2023

  1. Fix review comments 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    4f72d51 View commit details
    Browse the repository at this point in the history

  2. Fix review comments 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    d56d151 View commit details
    Browse the repository at this point in the history

Commits on Oct 19, 2023

  1. Fix review comments 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 19, 2023
    Configuration menu
    Copy the full SHA
    171915c View commit details
    Browse the repository at this point in the history

Commits on Oct 23, 2023

  1. Add sharding docs 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    24ee6ea View commit details
    Browse the repository at this point in the history

  2. Add sharding docs 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    aa8743c View commit details
    Browse the repository at this point in the history

Commits on Oct 24, 2023

  1. Merge remote-tracking branch 'nvidia/main' into add_jax_default_device

    @awolant
    awolant committed Oct 24, 2023
    Configuration menu
    Copy the full SHA
    781ff33 View commit details
    Browse the repository at this point in the history

  2. Add default device id to JAX decorated iterator fn 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 24, 2023
    Configuration menu
    Copy the full SHA
    745e3d5 View commit details
    Browse the repository at this point in the history

  3. Extend comments 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 24, 2023
    Configuration menu
    Copy the full SHA
    363b67e View commit details
    Browse the repository at this point in the history

  4. Add default args to JAX iterator 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 24, 2023
    Configuration menu
    Copy the full SHA
    293b66b View commit details
    Browse the repository at this point in the history

  5. Merge remote-tracking branch 'nvidia/main' into add_jax_default_device

    @awolant
    awolant committed Oct 24, 2023
    Configuration menu
    Copy the full SHA
    5d46d9e View commit details
    Browse the repository at this point in the history

  6. Remove unused stuff 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 24, 2023
    Configuration menu
    Copy the full SHA
    6ed2d32 View commit details
    Browse the repository at this point in the history

Commits on Oct 27, 2023

  1. Merge remote-tracking branch 'nvidia/main' into add_jax_default_device

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

  2. Add num_threads section 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 27, 2023
    Configuration menu
    Copy the full SHA
    013a441 View commit details
    Browse the repository at this point in the history

  3. Fix typo 

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    @awolant
    awolant committed Oct 27, 2023
    Configuration menu
    Copy the full SHA
    1b81b73 View commit details
    Browse the repository at this point in the history