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

feat: Flax #3123

Merged
merged 25 commits into from
Feb 23, 2023
Merged

feat: Flax #3123

merged 25 commits into from
Feb 23, 2023

Commits on Feb 21, 2023

  1. feat: flax skafold 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    58efa7c View commit details
    Browse the repository at this point in the history

  2. feat: Jax container 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    c8166ec View commit details
    Browse the repository at this point in the history

  3. feat: Flax 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    807b09b View commit details
    Browse the repository at this point in the history

  4. feat: add tests to example MNIST 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    8253a91 View commit details
    Browse the repository at this point in the history

  5. feat: adding flax tests 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    d8c9bfd View commit details
    Browse the repository at this point in the history

  6. fix: add tensorflow to Flax ci 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    4602267 View commit details
    Browse the repository at this point in the history

  7. revert: main 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    955f9f4 View commit details
    Browse the repository at this point in the history

  8. fix: rename equal_shape 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    320491d View commit details
    Browse the repository at this point in the history

  9. chore: export imports 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    fadb5e1 View commit details
    Browse the repository at this point in the history

  10. chore: address comments and convert payload to np.array 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    7651748 View commit details
    Browse the repository at this point in the history

  11. chore: enable plugins on CI only 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    5457a07 View commit details
    Browse the repository at this point in the history

  12. chore: add instruction to run standalone 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    a5eaed3 View commit details
    Browse the repository at this point in the history

  13. feat: adding hermetic options for training 

    (default to False)

Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    8dac69f View commit details
    Browse the repository at this point in the history

  14. fix: requirements 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    535d726 View commit details
    Browse the repository at this point in the history

  15. chore: add instruction for containerize 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    0396a27 View commit details
    Browse the repository at this point in the history

  16. chore: update readme 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    c681bb5 View commit details
    Browse the repository at this point in the history

  17. fix: tests 

    Signed-off-by: Aaron Pham <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    21863d7 View commit details
    Browse the repository at this point in the history

  18. chore: add gpu requirements 

    Signed-off-by: aarnphm-ec2-dev <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    934f1d1 View commit details
    Browse the repository at this point in the history

  19. fix: ignore user_defined_callback if it is a lambda function 

    Signed-off-by: aarnphm-ec2-dev <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    a947056 View commit details
    Browse the repository at this point in the history

  20. tests: run async_predict inside test 

    Signed-off-by: aarnphm-ec2-dev <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    b346a26 View commit details
    Browse the repository at this point in the history

  21. chore: revert changes from upstream/main 

    reverted changes: docs

Signed-off-by: Aaron <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    d227e6d View commit details
    Browse the repository at this point in the history

  22. chore: remove plasma client 

    Signed-off-by: Aaron <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    8155ba8 View commit details
    Browse the repository at this point in the history

  23. fix: typing 

    Signed-off-by: Aaron <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    5672420 View commit details
    Browse the repository at this point in the history

  24. chore: update tests 

    Signed-off-by: Aaron <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    e3c1021 View commit details
    Browse the repository at this point in the history

  25. chore: add API reference for Flax 

    Signed-off-by: Aaron <[email protected]>
    @aarnphm
    aarnphm committed Feb 21, 2023
    Configuration menu
    Copy the full SHA
    4e46247 View commit details
    Browse the repository at this point in the history