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

Introduce resources to streamline the combination of jax transformation with OOP pattern #44

Merged
merged 13 commits into from
Oct 9, 2023
Merged

Introduce resources to streamline the combination of jax transformation with OOP pattern #44

merged 13 commits into from
Oct 9, 2023

Commits on Aug 4, 2023

  1. Revert "Perform a deep copy when copying pytrees" 

    This reverts commit f8b5478.
    @diegoferigo
    diegoferigo committed Aug 4, 2023
    Configuration menu
    Copy the full SHA
    f1fbe68 View commit details
    Browse the repository at this point in the history

  2. Always restore mutability before exiting the context manager

    @diegoferigo
    diegoferigo committed Aug 4, 2023
    Configuration menu
    Copy the full SHA
    825ff5a View commit details
    Browse the repository at this point in the history

  3. Convert utils.py module to jaxsim.utils package

    @diegoferigo
    diegoferigo committed Aug 4, 2023
    Configuration menu
    Copy the full SHA
    719eb40 View commit details
    Browse the repository at this point in the history

  4. Create abstract class that can be automatically vectorized with vmap

    @diegoferigo
    diegoferigo committed Aug 4, 2023
    Configuration menu
    Copy the full SHA
    68ac4b5 View commit details
    Browse the repository at this point in the history

Commits on Oct 9, 2023

  1. Add decorators to apply jit and vmap on jax dataclasses methods

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    b005b8a View commit details
    Browse the repository at this point in the history

  2. Add tests for applying jax transformations in OOP on jax dataclasses

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    d922ccf View commit details
    Browse the repository at this point in the history

  3. Add typing_extension dependency for Python < 3.11

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    ce844c7 View commit details
    Browse the repository at this point in the history

  4. Use typing.Self in jaxsim_dataclass

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    d93b4ca View commit details
    Browse the repository at this point in the history

  5. Initialize __mutability__ in jaxsim_dataclass

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    a3bc673 View commit details
    Browse the repository at this point in the history

  6. Restore the original pytree content if mutable_context raises

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    84f5ef5 View commit details
    Browse the repository at this point in the history

  7. Extend the tracing detection checking also the Tracer type

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    6b424bf View commit details
    Browse the repository at this point in the history

  8. Remove resources deprecated in upstream

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    75b8d27 View commit details
    Browse the repository at this point in the history

  9. Run tests in subprocess and move pytest configuration to pyproject.toml

    @diegoferigo
    diegoferigo committed Oct 9, 2023
    Configuration menu
    Copy the full SHA
    c1f5b63 View commit details
    Browse the repository at this point in the history