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

Macenko JAX backend support #36

Open
wants to merge 16 commits into
base: development
Choose a base branch
from
Open

Macenko JAX backend support #36

wants to merge 16 commits into from

Commits on Jan 2, 2023

  1. added jax backend support for macenko

    @andreped
    andreped committed Jan 2, 2023
    Configuration menu
    Copy the full SHA
    b297a40 View commit details
    Browse the repository at this point in the history

Commits on Jan 3, 2023

  1. fixed bug in jax macenko

    @andreped
    andreped committed Jan 3, 2023
    Configuration menu
    Copy the full SHA
    07c2c58 View commit details
    Browse the repository at this point in the history

  2. fixed masking bug in jax macenko

    @andreped
    andreped committed Jan 3, 2023
    Configuration menu
    Copy the full SHA
    a20c69d View commit details
    Browse the repository at this point in the history

Commits on Jan 22, 2023

  1. made Jax Macenko class a PyTree -> Jax compatible [no ci]

    @andreped
    andreped committed Jan 22, 2023
    Configuration menu
    Copy the full SHA
    2366cc5 View commit details
    Browse the repository at this point in the history

  2. minor bug in _tree_flatten [no ci]

    @andreped
    andreped committed Jan 22, 2023
    Configuration menu
    Copy the full SHA
    4eddfc9 View commit details
    Browse the repository at this point in the history

Commits on Jan 23, 2023

  1. added macenko jax unit test

    @andreped
    andreped committed Jan 23, 2023
    Configuration menu
    Copy the full SHA
    4890bdb View commit details
    Browse the repository at this point in the history

  2. run CI on all branches + added jax tests to CI

    @andreped
    andreped committed Jan 23, 2023
    Configuration menu
    Copy the full SHA
    d84110f View commit details
    Browse the repository at this point in the history

  3. build full only on main + build quick on all branches

    @andreped
    andreped committed Jan 23, 2023
    Configuration menu
    Copy the full SHA
    8013108 View commit details
    Browse the repository at this point in the history

  4. renamed name in workflows

    @andreped
    andreped committed Jan 23, 2023
    Configuration menu
    Copy the full SHA
    cae7f95 View commit details
    Browse the repository at this point in the history

  5. fixed np.float issue + minor array refactor

    @andreped
    andreped committed Jan 23, 2023
    Configuration menu
    Copy the full SHA
    1877664 View commit details
    Browse the repository at this point in the history

Commits on Jan 29, 2023

  1. fix: setting jax.jit decorators differently yields correct output

    @andreped
    andreped committed Jan 29, 2023
    Configuration menu
    Copy the full SHA
    79b91aa View commit details
    Browse the repository at this point in the history

  2. fix: removed unwanted preprocessing in jax unit test

    @andreped
    andreped committed Jan 29, 2023
    Configuration menu
    Copy the full SHA
    dd3e8e6 View commit details
    Browse the repository at this point in the history

  3. refactor: updated README regarding JAX - match development branch

    @andreped
    andreped committed Jan 29, 2023
    Configuration menu
    Copy the full SHA
    9323c03 View commit details
    Browse the repository at this point in the history

  4. fix: updated setup.py to properly support jax during install

    @andreped
    andreped committed Jan 29, 2023
    Configuration menu
    Copy the full SHA
    b54c228 View commit details
    Browse the repository at this point in the history

  5. fix: run jax quick test CI on latest ubuntu + python 3.7

    @andreped
    andreped committed Jan 29, 2023
    Configuration menu
    Copy the full SHA
    50f0fc1 View commit details
    Browse the repository at this point in the history

  6. Merge branch 'development' into jax-backend

    @andreped
    andreped authored Jan 29, 2023
    Configuration menu
    Copy the full SHA
    dde3db4 View commit details
    Browse the repository at this point in the history