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

PyTorch Softmax Ops #846

Merged
merged 51 commits into from
Jun 28, 2024
Merged

PyTorch Softmax Ops #846

merged 51 commits into from
Jun 28, 2024

Commits on May 13, 2024

  1. Add pytorch support for some basic Ops

    @HarshvirSandhu
    HarshvirSandhu committed May 13, 2024
    Configuration menu
    Copy the full SHA
    27e2526 View commit details
    Browse the repository at this point in the history

  2. update variable names, docstrings

    @HarshvirSandhu
    HarshvirSandhu committed May 13, 2024
    Configuration menu
    Copy the full SHA
    629d00b View commit details
    Browse the repository at this point in the history

Commits on May 17, 2024

  1. Avoid numpy conversion of torch Tensors

    @HarshvirSandhu
    HarshvirSandhu committed May 17, 2024
    Configuration menu
    Copy the full SHA
    3eceb56 View commit details
    Browse the repository at this point in the history

  2. Fix typify and CheckAndRaise

    @HarshvirSandhu
    HarshvirSandhu committed May 17, 2024
    Configuration menu
    Copy the full SHA
    3cde964 View commit details
    Browse the repository at this point in the history

  3. Fix Elemwise Ops

    @HarshvirSandhu
    HarshvirSandhu committed May 17, 2024
    Configuration menu
    Copy the full SHA
    c003aa5 View commit details
    Browse the repository at this point in the history

  4. Fix Scalar Ops

    @HarshvirSandhu
    HarshvirSandhu committed May 17, 2024
    Configuration menu
    Copy the full SHA
    8dc406e View commit details
    Browse the repository at this point in the history

  5. Fix ruff-format

    @HarshvirSandhu
    HarshvirSandhu committed May 17, 2024
    Configuration menu
    Copy the full SHA
    a8f6ddb View commit details
    Browse the repository at this point in the history

Commits on May 23, 2024

  1. Initial setup for pytorch tests

    @HarshvirSandhu
    HarshvirSandhu committed May 23, 2024
    Configuration menu
    Copy the full SHA
    9d535f5 View commit details
    Browse the repository at this point in the history

  2. Fix mode parameters for pytorch

    @HarshvirSandhu
    HarshvirSandhu committed May 23, 2024
    Configuration menu
    Copy the full SHA
    c5600da View commit details
    Browse the repository at this point in the history

  3. Prevent conversion of scalars to numpy

    @HarshvirSandhu
    HarshvirSandhu committed May 23, 2024
    Configuration menu
    Copy the full SHA
    54b6248 View commit details
    Browse the repository at this point in the history

  4. Update TensorConstantSignature and map dtypes to Tensor types

    @HarshvirSandhu
    HarshvirSandhu committed May 23, 2024
    Configuration menu
    Copy the full SHA
    19454b3 View commit details
    Browse the repository at this point in the history

  5. Add tests for basic ops

    @HarshvirSandhu
    HarshvirSandhu committed May 23, 2024
    Configuration menu
    Copy the full SHA
    92d7114 View commit details
    Browse the repository at this point in the history

Commits on May 29, 2024

  1. Remove torch from user facing API

    @HarshvirSandhu
    HarshvirSandhu committed May 29, 2024
    Configuration menu
    Copy the full SHA
    5aae0e5 View commit details
    Browse the repository at this point in the history

  2. Add function to convert numpy arrays to pytorch tensors

    @HarshvirSandhu
    HarshvirSandhu committed May 29, 2024
    Configuration menu
    Copy the full SHA
    8c174dd View commit details
    Browse the repository at this point in the history

  3. Avoid copy when converting to tensor

    @HarshvirSandhu
    HarshvirSandhu committed May 29, 2024
    Configuration menu
    Copy the full SHA
    0977c3a View commit details
    Browse the repository at this point in the history

  4. Fix tests

    @HarshvirSandhu
    HarshvirSandhu committed May 29, 2024
    Configuration menu
    Copy the full SHA
    1c23825 View commit details
    Browse the repository at this point in the history

Commits on May 31, 2024

  1. Remove dispatches that are not tested

    @HarshvirSandhu
    HarshvirSandhu committed May 31, 2024
    Configuration menu
    Copy the full SHA
    c9195a8 View commit details
    Browse the repository at this point in the history

  2. set path for pytorch tests

    @HarshvirSandhu
    HarshvirSandhu committed May 31, 2024
    Configuration menu
    Copy the full SHA
    b07805c View commit details
    Browse the repository at this point in the history

Commits on Jun 4, 2024

  1. Remove tensorflow probability from yml

    @HarshvirSandhu
    HarshvirSandhu committed Jun 4, 2024
    Configuration menu
    Copy the full SHA
    9e8d3fc View commit details
    Browse the repository at this point in the history

  2. Add checks for runtime broadcasting

    @HarshvirSandhu
    HarshvirSandhu committed Jun 4, 2024
    Configuration menu
    Copy the full SHA
    a2d3afa View commit details
    Browse the repository at this point in the history

  3. Remove IfElse

    @HarshvirSandhu
    HarshvirSandhu committed Jun 4, 2024
    Configuration menu
    Copy the full SHA
    a577a80 View commit details
    Browse the repository at this point in the history

Commits on Jun 12, 2024

  1. Remove dev notebook

    @HarshvirSandhu
    HarshvirSandhu committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    499a174 View commit details
    Browse the repository at this point in the history

  2. Fix check and raise

    @HarshvirSandhu
    HarshvirSandhu committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    2826613 View commit details
    Browse the repository at this point in the history

  3. Fix compare_pytorch_and_py

    @HarshvirSandhu
    HarshvirSandhu committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    62ffcec View commit details
    Browse the repository at this point in the history

  4. Fix DimShuffle

    @HarshvirSandhu
    HarshvirSandhu committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    acdbba1 View commit details
    Browse the repository at this point in the history

  5. Add tests for Elemwise operations

    @HarshvirSandhu
    HarshvirSandhu committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    2519c65 View commit details
    Browse the repository at this point in the history

Commits on Jun 14, 2024

  1. Fix test for CheckAndRaise

    @HarshvirSandhu
    HarshvirSandhu committed Jun 14, 2024
    Configuration menu
    Copy the full SHA
    eb6d5c2 View commit details
    Browse the repository at this point in the history

  2. Remove duplicate function

    @HarshvirSandhu
    HarshvirSandhu committed Jun 14, 2024
    Configuration menu
    Copy the full SHA
    9f02a4f View commit details
    Browse the repository at this point in the history

Commits on Jun 15, 2024

  1. Remove device from pytorch_typify

    @HarshvirSandhu
    HarshvirSandhu committed Jun 15, 2024
    Configuration menu
    Copy the full SHA
    caf2965 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' of https://github.com/HarshvirSandhu/pytensor int… 

    …o pytensor-pytorch
    @HarshvirSandhu
    HarshvirSandhu committed Jun 15, 2024
    Configuration menu
    Copy the full SHA
    bf87eb9 View commit details
    Browse the repository at this point in the history

  3. Solve merge conflict

    @HarshvirSandhu
    HarshvirSandhu committed Jun 15, 2024
    Configuration menu
    Copy the full SHA
    2c27683 View commit details
    Browse the repository at this point in the history

  4. Use micromamba for pytorch install

    @HarshvirSandhu
    HarshvirSandhu committed Jun 15, 2024
    Configuration menu
    Copy the full SHA
    c603c6b View commit details
    Browse the repository at this point in the history

Commits on Jun 16, 2024

  1. Fix pytorch linker

    @HarshvirSandhu
    HarshvirSandhu committed Jun 16, 2024
    Configuration menu
    Copy the full SHA
    3f17107 View commit details
    Browse the repository at this point in the history

  2. Fix typify and deepcopy

    @HarshvirSandhu
    HarshvirSandhu committed Jun 16, 2024
    Configuration menu
    Copy the full SHA
    e850d8d View commit details
    Browse the repository at this point in the history

  3. Parametrize device in all tests

    @HarshvirSandhu
    HarshvirSandhu committed Jun 16, 2024
    Configuration menu
    Copy the full SHA
    e682fc4 View commit details
    Browse the repository at this point in the history

  4. Install torch with cuda

    @HarshvirSandhu
    HarshvirSandhu committed Jun 16, 2024
    Configuration menu
    Copy the full SHA
    bf4cf92 View commit details
    Browse the repository at this point in the history

  5. Fix test_pytorch_FunctionGraph_once

    @HarshvirSandhu
    HarshvirSandhu committed Jun 16, 2024
    Configuration menu
    Copy the full SHA
    899e7f9 View commit details
    Browse the repository at this point in the history

  6. Remove device argument from test

    @HarshvirSandhu
    HarshvirSandhu committed Jun 16, 2024
    Configuration menu
    Copy the full SHA
    04d2935 View commit details
    Browse the repository at this point in the history

Commits on Jun 17, 2024

  1. remove device from elemwise tests and add assertions

    @HarshvirSandhu
    HarshvirSandhu committed Jun 17, 2024
    Configuration menu
    Copy the full SHA
    8ec7661 View commit details
    Browse the repository at this point in the history

  2. skip tests if cuda is not available

    @HarshvirSandhu
    HarshvirSandhu committed Jun 17, 2024
    Configuration menu
    Copy the full SHA
    bb7df41 View commit details
    Browse the repository at this point in the history

Commits on Jun 18, 2024

  1. Fix tests

    @HarshvirSandhu
    HarshvirSandhu committed Jun 18, 2024
    Configuration menu
    Copy the full SHA
    0441cf2 View commit details
    Browse the repository at this point in the history

Commits on Jun 20, 2024

  1. Merge branch 'main' of https://github.com/pymc-devs/pytensor into pyt… 

    …ensor-pytorch-softmax
    @HAKSOAT
    HAKSOAT committed Jun 20, 2024
    Configuration menu
    Copy the full SHA
    85f2742 View commit details
    Browse the repository at this point in the history

Commits on Jun 23, 2024

  1. Implemented softmax ops for PyTorch

    @HAKSOAT
    HAKSOAT committed Jun 23, 2024
    Configuration menu
    Copy the full SHA
    4ca5aca View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'upstream/main' into pytensor-pytorch-so… 

    …ftmax
    @HAKSOAT
    HAKSOAT committed Jun 23, 2024
    Configuration menu
    Copy the full SHA
    b9aca57 View commit details
    Browse the repository at this point in the history

Commits on Jun 24, 2024

  1. Switched to run softmax on all items if axis is None

    @HAKSOAT
    HAKSOAT committed Jun 24, 2024
    Configuration menu
    Copy the full SHA
    287d9c2 View commit details
    Browse the repository at this point in the history

  2. Implemented log softmax

    @HAKSOAT
    HAKSOAT committed Jun 24, 2024
    Configuration menu
    Copy the full SHA
    f42e2a0 View commit details
    Browse the repository at this point in the history

Commits on Jun 25, 2024

  1. Implemented softmaxgrad

    @HAKSOAT
    HAKSOAT committed Jun 25, 2024
    Configuration menu
    Copy the full SHA
    35b17e0 View commit details
    Browse the repository at this point in the history

Commits on Jun 27, 2024

  1. Added checks and error raises for nonfloat inputs

    @HAKSOAT
    HAKSOAT committed Jun 27, 2024
    Configuration menu
    Copy the full SHA
    5efc3c8 View commit details
    Browse the repository at this point in the history

Commits on Jun 28, 2024

  1. Added checks and error raises for nonfloat inputs

    @HAKSOAT
    HAKSOAT committed Jun 28, 2024
    Configuration menu
    Copy the full SHA
    16e415a View commit details
    Browse the repository at this point in the history

  2. Merge branch 'pytensor-pytorch-softmax' of https://github.com/HAKSOAT… 

    …/pytensor into pytensor-pytorch-softmax
    @HAKSOAT
    HAKSOAT committed Jun 28, 2024
    Configuration menu
    Copy the full SHA
    ffbc594 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' of https://github.com/pymc-devs/pytensor into pyt… 

    …ensor-pytorch-softmax
    @HAKSOAT
    HAKSOAT committed Jun 28, 2024
    Configuration menu
    Copy the full SHA
    b4cdce0 View commit details
    Browse the repository at this point in the history