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

Auto compile Lux models to reactant #665

Draft
wants to merge 15 commits into
base: main
Choose a base branch
from
Draft

Auto compile Lux models to reactant #665

wants to merge 15 commits into from

Commits on Jun 1, 2024

  1. Auto compile Lux models to reactant

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    405aa42 View commit details
    Browse the repository at this point in the history

  2. Work around the returning states issue

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    e544690 View commit details
    Browse the repository at this point in the history

  3. Make the states type stable

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    018d7a0 View commit details
    Browse the repository at this point in the history

  4. Try using Enzyme for the backward pass

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    7ec54e7 View commit details
    Browse the repository at this point in the history

  5. Add a force backward compile mode

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    2414d99 View commit details
    Browse the repository at this point in the history

  6. Special handling for mixed eltypes in Reactant

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    a7bf7a6 View commit details
    Browse the repository at this point in the history

  7. Add checks to validate matching structure

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    7ea236f View commit details
    Browse the repository at this point in the history

  8. Try converting common parameter types to the compiled type

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    d186bf6 View commit details
    Browse the repository at this point in the history

  9. Accidentally used Boxed value

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    c632418 View commit details
    Browse the repository at this point in the history

  10. Implement a working VJP function

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    f5a3e58 View commit details
    Browse the repository at this point in the history

  11. Move things around

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    55d4767 View commit details
    Browse the repository at this point in the history

  12. Training step function

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    db3949e View commit details
    Browse the repository at this point in the history

  13. Start impl for Reactant training loop

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    7ec09d7 View commit details
    Browse the repository at this point in the history

  14. More comprehensive compilation options

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    916e685 View commit details
    Browse the repository at this point in the history

  15. Allow skipping all compilation

    @avik-pal
    avik-pal committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    8ce9707 View commit details
    Browse the repository at this point in the history