Skip to content
This repository has been archived by the owner on May 11, 2023. It is now read-only.

Parameter state #21

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

Parameter state #21

wants to merge 84 commits into from

Commits on Feb 7, 2023

  1. (1) Remove PyTree.

    @daniel-dodd
    daniel-dodd committed Feb 7, 2023
    Configuration menu
    Copy the full SHA
    b505f56 View commit details
    Browse the repository at this point in the history

  2. remove dict

    @daniel-dodd
    daniel-dodd committed Feb 7, 2023
    Configuration menu
    Copy the full SHA
    bdcfacb View commit details
    Browse the repository at this point in the history

  3. Add stuff.

    @daniel-dodd
    daniel-dodd committed Feb 7, 2023
    Configuration menu
    Copy the full SHA
    21b2619 View commit details
    Browse the repository at this point in the history

  4. Update structure.

    @daniel-dodd
    daniel-dodd committed Feb 7, 2023
    Configuration menu
    Copy the full SHA
    ca6b5a3 View commit details
    Browse the repository at this point in the history

  5. Update README.md

    @daniel-dodd
    daniel-dodd committed Feb 7, 2023
    Configuration menu
    Copy the full SHA
    cf76b0a View commit details
    Browse the repository at this point in the history

Commits on Feb 8, 2023

  1. Remove base. Move to module.

    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    e3640db View commit details
    Browse the repository at this point in the history

  2. Update README.md

    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    a7f4f65 View commit details
    Browse the repository at this point in the history

  3. Remove config. Drop distrax.

    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    2aa295c View commit details
    Browse the repository at this point in the history

  4. Update README.md

    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    1c22a87 View commit details
    Browse the repository at this point in the history

  5. Update README.md

    @daniel-dodd
    daniel-dodd authored Feb 8, 2023
    Configuration menu
    Copy the full SHA
    1cc7b54 View commit details
    Browse the repository at this point in the history

  6. Refactor.

    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    7438478 View commit details
    Browse the repository at this point in the history

  7. Merge branch 'equinox' of https://github.com/JaxGaussianProcesses/Jax… 

    …Utils into equinox
    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    8518e61 View commit details
    Browse the repository at this point in the history

  8. Create test_bijectors.py

    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    fc8010e View commit details
    Browse the repository at this point in the history

  9. Create test_objective.py

    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    768716c View commit details
    Browse the repository at this point in the history

  10. Update.

    @daniel-dodd
    daniel-dodd committed Feb 8, 2023
    Configuration menu
    Copy the full SHA
    6b1029a View commit details
    Browse the repository at this point in the history

Commits on Feb 9, 2023

  1. Test module.

    @daniel-dodd
    daniel-dodd committed Feb 9, 2023
    Configuration menu
    Copy the full SHA
    a6d2d45 View commit details
    Browse the repository at this point in the history

  2. Update setup.py

    @daniel-dodd
    daniel-dodd committed Feb 9, 2023
    Configuration menu
    Copy the full SHA
    a2c6560 View commit details
    Browse the repository at this point in the history

  3. Update setup.py

    @daniel-dodd
    daniel-dodd committed Feb 9, 2023
    Configuration menu
    Copy the full SHA
    857cf8b View commit details
    Browse the repository at this point in the history

  4. Update setup.py

    @daniel-dodd
    daniel-dodd committed Feb 9, 2023
    Configuration menu
    Copy the full SHA
    9ba5451 View commit details
    Browse the repository at this point in the history

  5. Update setup.py

    @daniel-dodd
    daniel-dodd committed Feb 9, 2023
    Configuration menu
    Copy the full SHA
    d5bf42d View commit details
    Browse the repository at this point in the history

  6. Update fit.py

    @daniel-dodd
    daniel-dodd committed Feb 9, 2023
    Configuration menu
    Copy the full SHA
    7646a0f View commit details
    Browse the repository at this point in the history

  7. Delete params.py

    @daniel-dodd
    daniel-dodd committed Feb 9, 2023
    Configuration menu
    Copy the full SHA
    e9eb8d2 View commit details
    Browse the repository at this point in the history

Commits on Feb 12, 2023

  1. Update.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    6f3f63d View commit details
    Browse the repository at this point in the history

  2. Add test.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    4d037a6 View commit details
    Browse the repository at this point in the history

  3. Update.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    4aee517 View commit details
    Browse the repository at this point in the history

  4. Update module.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    02c5fa1 View commit details
    Browse the repository at this point in the history

  5. Update module.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    509cf19 View commit details
    Browse the repository at this point in the history

  6. Update fit.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    63cad7f View commit details
    Browse the repository at this point in the history

  7. Add some checks to fit.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    026d045 View commit details
    Browse the repository at this point in the history

  8. Update checks.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    2765c30 View commit details
    Browse the repository at this point in the history

  9. Add default trainable status to param field.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    d36ad74 View commit details
    Browse the repository at this point in the history

  10. Update module.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    fad45c0 View commit details
    Browse the repository at this point in the history

  11. Update module.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    0fb7b8d View commit details
    Browse the repository at this point in the history

  12. Update bijectors.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    d1a2d3d View commit details
    Browse the repository at this point in the history

  13. Update bijectors.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    3089fcb View commit details
    Browse the repository at this point in the history

  14. Remove legacy verify dataset.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    fad89c6 View commit details
    Browse the repository at this point in the history

  15. Update fit.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    0737314 View commit details
    Browse the repository at this point in the history

  16. Add compilation message to progress bar.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    daf5d22 View commit details
    Browse the repository at this point in the history

  17. Leaf node checking.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    15fce76 View commit details
    Browse the repository at this point in the history

  18. Update module.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    4253485 View commit details
    Browse the repository at this point in the history

  19. Switch from functional to leaf approach

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    51b33ff View commit details
    Browse the repository at this point in the history

  20. Switch to metadata.

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    7f4a7f6 View commit details
    Browse the repository at this point in the history

  21. Add minimal verbose scan. Need to generalise and cleanup!

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    580a36c View commit details
    Browse the repository at this point in the history

  22. Update module.py

    @daniel-dodd
    daniel-dodd committed Feb 12, 2023
    Configuration menu
    Copy the full SHA
    29d44f6 View commit details
    Browse the repository at this point in the history

Commits on Feb 14, 2023

  1. Fix jaxutils.fit __init__.py import, add documentation to `module… 

    ….py`
    @daniel-dodd
    daniel-dodd committed Feb 14, 2023
    Configuration menu
    Copy the full SHA
    79dace9 View commit details
    Browse the repository at this point in the history

Commits on Feb 17, 2023

  1. Bugfix.

    @daniel-dodd
    daniel-dodd committed Feb 17, 2023
    Configuration menu
    Copy the full SHA
    c2cf3c2 View commit details
    Browse the repository at this point in the history

Commits on Feb 19, 2023

  1. Update.

    @daniel-dodd
    daniel-dodd committed Feb 19, 2023
    Configuration menu
    Copy the full SHA
    fcdd52e View commit details
    Browse the repository at this point in the history

  2. Update module.py

    @daniel-dodd
    daniel-dodd committed Feb 19, 2023
    Configuration menu
    Copy the full SHA
    6971942 View commit details
    Browse the repository at this point in the history

  3. Fix list tuple breakages (SEE NOTE). 

    TODO:

- PEFORMANCE: Refactor metadata to make it easier to set, modify, access. `module.at[...].set_meta(**kwargs), could raise error / do some type checking if keyword is e.g., not in metadata, but maybe this might add greater flexibility. It would be better to access trainables, bijectors, etc direcly from a metadata collection.

BUG: Need to reintroduce checks / create a policy if a field is not marked with jaxutils.param, to determine whether a field is a node or not.
    @daniel-dodd
    daniel-dodd committed Feb 19, 2023
    Configuration menu
    Copy the full SHA
    3767684 View commit details
    Browse the repository at this point in the history

Commits on Feb 20, 2023

  1. Minimal functionality for accessing and setting trainables, transform… 

    …s and priors.
    @daniel-dodd
    daniel-dodd committed Feb 20, 2023
    Configuration menu
    Copy the full SHA
    f753734 View commit details
    Browse the repository at this point in the history

Commits on Feb 22, 2023

  1. Push minimal meta functionality.

    @daniel-dodd
    daniel-dodd committed Feb 22, 2023
    Configuration menu
    Copy the full SHA
    9997805 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'equinox' of https://github.com/JaxGaussianProcesses/Jax… 

    …Utils into equinox
    @daniel-dodd
    daniel-dodd committed Feb 22, 2023
    Configuration menu
    Copy the full SHA
    6a6c244 View commit details
    Browse the repository at this point in the history

  3. Python 3.8 does not like "|"

    @daniel-dodd
    daniel-dodd committed Feb 22, 2023
    Configuration menu
    Copy the full SHA
    ce52640 View commit details
    Browse the repository at this point in the history

  4. Update test_module.py

    @daniel-dodd
    daniel-dodd committed Feb 22, 2023
    Configuration menu
    Copy the full SHA
    c289133 View commit details
    Browse the repository at this point in the history

  5. Update test_module.py

    @daniel-dodd
    daniel-dodd committed Feb 22, 2023
    Configuration menu
    Copy the full SHA
    a47c8dd View commit details
    Browse the repository at this point in the history

Commits on Mar 7, 2023

  1. Redesign PyTree and introduce Module. 

    For transparency this is heavily inspired by `Equinox` and `Simple-Pytree`, which have been adapted to provide metadata functionality, to apply transformations and stop gradients, etc. We hope to be Equinox compatible, as much as possible.
    @daniel-dodd
    daniel-dodd committed Mar 7, 2023
    Configuration menu
    Copy the full SHA
    08920bf View commit details
    Browse the repository at this point in the history

Commits on Mar 8, 2023

  1. Remove some redundant code.

    @daniel-dodd
    daniel-dodd committed Mar 8, 2023
    Configuration menu
    Copy the full SHA
    75574d9 View commit details
    Browse the repository at this point in the history

Commits on Mar 9, 2023

  1. ParameterState.

    @daniel-dodd
    daniel-dodd committed Mar 9, 2023
    Configuration menu
    Copy the full SHA
    731f46e View commit details
    Browse the repository at this point in the history

  2. Add stuff to __init__

    @daniel-dodd
    daniel-dodd committed Mar 9, 2023
    Configuration menu
    Copy the full SHA
    3710e9c View commit details
    Browse the repository at this point in the history

  3. Minimal test for fit.

    @daniel-dodd
    daniel-dodd committed Mar 9, 2023
    Configuration menu
    Copy the full SHA
    aadf1d0 View commit details
    Browse the repository at this point in the history

  4. Update fit tests.

    @daniel-dodd
    daniel-dodd committed Mar 9, 2023
    Configuration menu
    Copy the full SHA
    7b63890 View commit details
    Browse the repository at this point in the history

  5. Update setup.py

    @daniel-dodd
    daniel-dodd committed Mar 9, 2023
    Configuration menu
    Copy the full SHA
    c95114e View commit details
    Browse the repository at this point in the history

  6. Add basic dict methods

    @thomaspinder
    thomaspinder committed Mar 9, 2023
    Configuration menu
    Copy the full SHA
    2a67a50 View commit details
    Browse the repository at this point in the history

Commits on Mar 10, 2023

  1. Add priors

    @thomaspinder
    thomaspinder committed Mar 10, 2023
    Configuration menu
    Copy the full SHA
    49559a4 View commit details
    Browse the repository at this point in the history

  2. Add log prior density fn

    @thomaspinder
    thomaspinder committed Mar 10, 2023
    Configuration menu
    Copy the full SHA
    fd97b43 View commit details
    Browse the repository at this point in the history

  3. Add unit test

    @thomaspinder
    thomaspinder committed Mar 10, 2023
    Configuration menu
    Copy the full SHA
    37bb0ee View commit details
    Browse the repository at this point in the history

  4. Merge pull request #18 from JaxGaussianProcesses/priors 

    Add priors
    @thomaspinder
    thomaspinder authored Mar 10, 2023
    Configuration menu
    Copy the full SHA
    00109f7 View commit details
    Browse the repository at this point in the history

Commits on Mar 13, 2023

  1. Parameters unit tests

    @thomaspinder
    thomaspinder committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    9f7a32a View commit details
    Browse the repository at this point in the history

  2. Test key-structure for bijectors and distributions

    @thomaspinder
    thomaspinder committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    33a7c6a View commit details
    Browse the repository at this point in the history

  3. Test key-structure for bijectors and distributions

    @thomaspinder
    thomaspinder committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    de692bf View commit details
    Browse the repository at this point in the history

  4. Remove npm-files

    @thomaspinder
    thomaspinder committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    1c33661 View commit details
    Browse the repository at this point in the history

  5. Action TODOs

    @thomaspinder
    thomaspinder committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    92c9396 View commit details
    Browse the repository at this point in the history

  6. Lint and format files

    @thomaspinder
    thomaspinder committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    dba579c View commit details
    Browse the repository at this point in the history

Commits on Mar 14, 2023

  1. Add combine fn.

    @thomaspinder
    thomaspinder committed Mar 14, 2023
    Configuration menu
    Copy the full SHA
    c506a20 View commit details
    Browse the repository at this point in the history

  2. Add parameter method

    @thomaspinder
    thomaspinder committed Mar 14, 2023
    Configuration menu
    Copy the full SHA
    57f46d7 View commit details
    Browse the repository at this point in the history

Commits on Mar 15, 2023

  1. Augment add_parameter

    @thomaspinder
    thomaspinder committed Mar 15, 2023
    Configuration menu
    Copy the full SHA
    9b16787 View commit details
    Browse the repository at this point in the history

Commits on Mar 19, 2023

  1. simplify fit

    @thomaspinder
    thomaspinder committed Mar 19, 2023
    Configuration menu
    Copy the full SHA
    5698dd7 View commit details
    Browse the repository at this point in the history

  2. simplify fit

    @thomaspinder
    thomaspinder committed Mar 19, 2023
    Configuration menu
    Copy the full SHA
    d91c6fb View commit details
    Browse the repository at this point in the history

  3. Update tests

    @thomaspinder
    thomaspinder committed Mar 19, 2023
    Configuration menu
    Copy the full SHA
    eaae509 View commit details
    Browse the repository at this point in the history

Commits on Mar 23, 2023

  1. Fix failing test

    @thomaspinder
    thomaspinder committed Mar 23, 2023
    Configuration menu
    Copy the full SHA
    ee7cb47 View commit details
    Browse the repository at this point in the history

  2. Fix failing test

    @thomaspinder
    thomaspinder committed Mar 23, 2023
    Configuration menu
    Copy the full SHA
    a4a2fa1 View commit details
    Browse the repository at this point in the history

  3. Fix failing test

    @thomaspinder
    thomaspinder committed Mar 23, 2023
    Configuration menu
    Copy the full SHA
    f8d2288 View commit details
    Browse the repository at this point in the history

  4. Switch PyTree version

    @thomaspinder
    thomaspinder committed Mar 23, 2023
    Configuration menu
    Copy the full SHA
    ae153b2 View commit details
    Browse the repository at this point in the history

  5. Update workflow

    @thomaspinder
    thomaspinder committed Mar 23, 2023
    Configuration menu
    Copy the full SHA
    0d34df5 View commit details
    Browse the repository at this point in the history