Code

ML models#80

Code

Open

mbreyt wants to merge 16 commits intoins-amu:mainins-amu/vbjax:mainfrom mbreyt:ml_modelsmbreyt/vbjax:ml_modelsCopy head branch name to clipboard

+651-3

Commits

Commits on Jan 24, 2024

• create noise generator module

  

  martin breyton

  authored and

  martin breyton

  committed
  53f9583

  Copy full SHA for 53f9583

• add noise_generator import

  

  martin breyton

  authored and

  martin breyton

  committed
  1a64dc0

  Copy full SHA for 1a64dc0

• add documentation and remove useless args in PSDGenerator definition

  

  martin breyton

  authored and

  martin breyton

  committed
  05c352f

  Copy full SHA for 05c352f

Commits on Feb 6, 2024

• work up to training over etas commit

  

  martin breyton

  authored and

  martin breyton

  committed
  f3a7fd3

  Copy full SHA for f3a7fd3

Commits on Mar 19, 2024

• update structure

  

  martin breyton

  authored and

  martin breyton

  committed
  92b7cdd

  Copy full SHA for 92b7cdd

Commits on May 14, 2024

• add MADE network

  

  martin breyton

  authored and

  martin breyton

  committed
  33b99a7

  Copy full SHA for 33b99a7

• update from upstream

  

  martin breyton

  authored and

  martin breyton

  committed
  3a63528

  Copy full SHA for 3a63528

• back to normal

  

  martin breyton

  authored and

  martin breyton

  committed
  785f04f

  Copy full SHA for 785f04f

• add MADE

  

  martin breyton

  authored and

  martin breyton

  committed
  bbda1c4

  Copy full SHA for bbda1c4

Commits on May 15, 2024

• jaxify train and eval functions

  mbreyt

  committed
  10624e7

  Copy full SHA for 10624e7

• add hard coded init scale to kernel

  mbreyt

  committed
  a15fc28

  Copy full SHA for a15fc28

• add MAF implementation

  mbreyt

  committed
  d64bed0

  Copy full SHA for d64bed0

• add imports from train_utils and ml_models

  mbreyt

  committed
  90acdd7

  Copy full SHA for 90acdd7

• refactor training example

  mbreyt

  committed
  06c028c

  Copy full SHA for 06c028c

Commits on May 17, 2024

• fix typo in shape dimension

  mbreyt

  committed
  cc000c2

  Copy full SHA for cc000c2

Commits on Jun 14, 2024

• add models

  mbreyt

  committed
  d2b6db7

  Copy full SHA for d2b6db7