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

Move StateDict to Haliax, clean it up a lot. #90

Open
wants to merge 18 commits into
base: main
Choose a base branch
from
Open

Move StateDict to Haliax, clean it up a lot. #90

wants to merge 18 commits into from

Commits on May 16, 2024

  1. wip

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    0609b32 View commit details
    Browse the repository at this point in the history

  2. wip state_dict serialization for haliax

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    80890e8 View commit details
    Browse the repository at this point in the history

  3. wip

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    d361bf7 View commit details
    Browse the repository at this point in the history

  4. add flatten_linear_layers to Linear. probably gonna move it out though

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    0420a1a View commit details
    Browse the repository at this point in the history

  5. this is much cleaner

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    9e4e857 View commit details
    Browse the repository at this point in the history

  6. tests

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    f1d0c0f View commit details
    Browse the repository at this point in the history

  7. remove old bad jax config bug

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    68dfb87 View commit details
    Browse the repository at this point in the history

  8. oops

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    02fc32e View commit details
    Browse the repository at this point in the history

  9. ok, the new stacked seems to basically work

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    54c4bfc View commit details
    Browse the repository at this point in the history

  10. add an explicit to_torch_compatible dict

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    c7a61dc View commit details
    Browse the repository at this point in the history

  11. make a public model for state

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    382e405 View commit details
    Browse the repository at this point in the history

  12. wip

    @dlwh
    dlwh committed May 16, 2024
    Configuration menu
    Copy the full SHA
    9061cfb View commit details
    Browse the repository at this point in the history

Commits on Jun 1, 2024

  1. Merge remote-tracking branch 'origin/main' into state_dict_in_haliax

    @dlwh
    dlwh committed Jun 1, 2024
    Configuration menu
    Copy the full SHA
    b64a374 View commit details
    Browse the repository at this point in the history

Commits on Jun 9, 2024

  1. Merge remote-tracking branch 'origin/main' into state_dict_in_haliax

    @dlwh
    dlwh committed Jun 9, 2024
    Configuration menu
    Copy the full SHA
    1799fe8 View commit details
    Browse the repository at this point in the history

Commits on Jun 10, 2024

  1. add docs, clean up tests some

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

  2. fix deprecations

    @dlwh
    dlwh committed Jun 10, 2024
    Configuration menu
    Copy the full SHA
    0ee7136 View commit details
    Browse the repository at this point in the history

Commits on Aug 27, 2024

  1. Merge remote-tracking branch 'origin/main' into state_dict_in_haliax

    @dlwh
    dlwh committed Aug 27, 2024
    Configuration menu
    Copy the full SHA
    c6bc2de View commit details
    Browse the repository at this point in the history

  2. remove update_state_dict

    @dlwh
    dlwh committed Aug 27, 2024
    Configuration menu
    Copy the full SHA
    c877679 View commit details
    Browse the repository at this point in the history