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 wrapper eager_function #41

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Introduce wrapper eager_function #41

wants to merge 4 commits into from

Commits on Apr 23, 2021

  1. Introduce wrapper eager_function 

    Introduce as_tensors, as_tensors_, as_raw_tensor, as_raw_tensors that rely in tree_flatten/tree_unflatten for more generic usages
JaxTensor is no longer registered as a pytree datastructure
Refactor JaxTensor._value_and_grad_fn
    @eserie
    eserie committed Apr 23, 2021
    Configuration menu
    Copy the full SHA
    a95265c View commit details
    Browse the repository at this point in the history

Commits on Apr 25, 2021

  1. astensor manage non tensor arguments and behave like identity if the … 

    …argument is not a tensor
    @eserie
    eserie committed Apr 25, 2021
    Configuration menu
    Copy the full SHA
    81bba4d View commit details
    Browse the repository at this point in the history

  2. new function astensors_any 

    Register all Tensor types as pytrees.
Handle unwraping mechanism within unflatten method.
    @eserie
    eserie committed Apr 25, 2021
    Configuration menu
    Copy the full SHA
    3de8820 View commit details
    Browse the repository at this point in the history

Commits on Apr 26, 2021

  1. remove skip_argnums argument since as_tensors_any manage non pytree o… 

    …bjects
    @eserie
    eserie committed Apr 26, 2021
    Configuration menu
    Copy the full SHA
    4583260 View commit details
    Browse the repository at this point in the history