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

PET-JAX #82

Closed
wants to merge 19 commits into from
Closed

PET-JAX #82

wants to merge 19 commits into from

Commits on Feb 16, 2024

  1. Core implementation

    @frostedoyster
    frostedoyster committed Feb 16, 2024
    Configuration menu
    Copy the full SHA
    b93dbb2 View commit details
    Browse the repository at this point in the history

Commits on Feb 18, 2024

  1. Allow use of standard attention!

    @frostedoyster
    frostedoyster committed Feb 18, 2024
    Configuration menu
    Copy the full SHA
    9086243 View commit details
    Browse the repository at this point in the history

  2. Expose more hypers

    @frostedoyster
    frostedoyster committed Feb 18, 2024
    Configuration menu
    Copy the full SHA
    ea74f6a View commit details
    Browse the repository at this point in the history

Commits on Feb 19, 2024

  1. Attempted torch model

    @frostedoyster
    frostedoyster committed Feb 19, 2024
    Configuration menu
    Copy the full SHA
    ebd2a54 View commit details
    Browse the repository at this point in the history

Commits on Feb 23, 2024

  1. Fail

    @frostedoyster
    frostedoyster committed Feb 23, 2024
    Configuration menu
    Copy the full SHA
    8fa9450 View commit details
    Browse the repository at this point in the history

  2. Complete conversion to torch

    @frostedoyster
    frostedoyster committed Feb 23, 2024
    Configuration menu
    Copy the full SHA
    7a3e5bd View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' into pet-jax

    @frostedoyster
    frostedoyster authored Feb 23, 2024
    Configuration menu
    Copy the full SHA
    e7b447f View commit details
    Browse the repository at this point in the history

Commits on Feb 25, 2024

  1. Message passing

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    491c7ef View commit details
    Browse the repository at this point in the history

  2. Some more tests

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    a821d1d View commit details
    Browse the repository at this point in the history

  3. Remove stray default dtype

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    bcc100d View commit details
    Browse the repository at this point in the history

  4. Merge branch 'stray-dtype' into pet-jax

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    8e2b5d6 View commit details
    Browse the repository at this point in the history

  5. More test structure

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    a9cd955 View commit details
    Browse the repository at this point in the history

  6. Finalize torch version

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    7665e2c View commit details
    Browse the repository at this point in the history

  7. More tests

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    61ef2ca View commit details
    Browse the repository at this point in the history

  8. Add minimal documentation

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    ff1377b View commit details
    Browse the repository at this point in the history

  9. Fix tests?

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    77d5848 View commit details
    Browse the repository at this point in the history

  10. Fix tests

    @frostedoyster
    frostedoyster committed Feb 25, 2024
    Configuration menu
    Copy the full SHA
    34f7cad View commit details
    Browse the repository at this point in the history

Commits on Feb 26, 2024

  1. Merge branch 'main' into pet-jax

    @frostedoyster
    frostedoyster authored Feb 26, 2024
    Configuration menu
    Copy the full SHA
    9be1601 View commit details
    Browse the repository at this point in the history

  2. Fix tests???

    @frostedoyster
    frostedoyster committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    3b7880e View commit details
    Browse the repository at this point in the history