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

Out axis linear #104

Merged
merged 6 commits into from
Apr 8, 2024
Merged

Out axis linear #104

merged 6 commits into from
Apr 8, 2024

Commits on Apr 8, 2024

  1. non-int out_axis in linear 

    - enable more flexible linear layer by making both in_features and out_features accepting variable dim
    @ASEM000
    ASEM000 committed Apr 8, 2024
    Configuration menu
    Copy the full SHA
    9d2c99b View commit details
    Browse the repository at this point in the history

  2. fix linear failing

    @ASEM000
    ASEM000 committed Apr 8, 2024
    Configuration menu
    Copy the full SHA
    b6eb925 View commit details
    Browse the repository at this point in the history

  3. use nn.linear in MLP

    @ASEM000
    ASEM000 committed Apr 8, 2024
    Configuration menu
    Copy the full SHA
    c188143 View commit details
    Browse the repository at this point in the history

  4. nits

    @ASEM000
    ASEM000 committed Apr 8, 2024
    Configuration menu
    Copy the full SHA
    d9e92d3 View commit details
    Browse the repository at this point in the history

  5. Update linear.py

    @ASEM000
    ASEM000 committed Apr 8, 2024
    Configuration menu
    Copy the full SHA
    e3840a1 View commit details
    Browse the repository at this point in the history

  6. fix doctest fail

    @ASEM000
    ASEM000 committed Apr 8, 2024
    Configuration menu
    Copy the full SHA
    35fc478 View commit details
    Browse the repository at this point in the history