Code

Shallow ensembles#275

Code

Merged

M-R-Schaefer merged 38 commits intodevapax-hub/apax:devfrom shallow_ensapax-hub/apax:shallow_ensCopy head branch name to clipboardJul 9, 2024

+336-72

Commits

Commits on May 4, 2024

• correct ordering of labels and predictions

  M-R-Schaefer

  committed
  9cd4f6c

  Copy full SHA for 9cd4f6c

• added crps and nll loss functions

  M-R-Schaefer

  committed
  b0d6c9a

  Copy full SHA for b0d6c9a

• added shallow ens to config

  M-R-Schaefer

  committed
  084d2af

  Copy full SHA for 084d2af

• added possibility for shallow ens in readout

  M-R-Schaefer

  committed
  8e1b4d0

  Copy full SHA for 8e1b4d0

• added preliminary shallow ens class

  M-R-Schaefer

  committed
  359f037

  Copy full SHA for 359f037

• added shallow ens to builder

  M-R-Schaefer

  committed
  705f109

  Copy full SHA for 705f109

• fixed tests

  M-R-Schaefer

  committed
  1520ff5

  Copy full SHA for 1520ff5

Commits on May 7, 2024

• removed debug print statements

  M-R-Schaefer

  committed
  48dc3ed

  Copy full SHA for 48dc3ed

• correctly pass n_shallow ens in builder

  M-R-Schaefer

  committed
  df7bee5

  Copy full SHA for df7bee5

• remove debug print statements

  M-R-Schaefer

  committed
  c786668

  Copy full SHA for c786668

• docstring

  M-R-Schaefer

  committed
  c1f734a

  Copy full SHA for c1f734a

Commits on May 8, 2024

• return ensemble predictions from model

  M-R-Schaefer

  committed
  6a83516

  Copy full SHA for 6a83516

• Merge branch 'dev' into shallow_ens

  M-R-Schaefer

  committed
  b9a9ca1

  Copy full SHA for b9a9ca1

Commits on May 10, 2024

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 6e25069
  pre-commit-ci[bot]

  committed
  6e25069

  Copy full SHA for 6e25069

Commits on May 15, 2024

• implemented switch between mean force and ensemble force computation

  M-R-Schaefer

  committed
  d22dc6c

  Copy full SHA for d22dc6c

• Merge branch 'shallow_ens' of https://github.com/apax-hub/apax into shallow_ens

  M-R-Schaefer

  committed
  4fcdd8d

  Copy full SHA for 4fcdd8d

Commits on May 17, 2024

• implemented stress prediction for shallow ens

  M-R-Schaefer

  committed
  af81217

  Copy full SHA for af81217

• compatibility with BAL

  M-R-Schaefer

  committed
  df19d04

  Copy full SHA for df19d04

• loss fn divisor

  M-R-Schaefer

  committed
  7f51cdd

  Copy full SHA for 7f51cdd

Commits on May 22, 2024

• added ApaxImport node

  M-R-Schaefer

  committed
  d17a8bf

  Copy full SHA for d17a8bf

Commits on Jun 1, 2024

• made import node compatible with jaxmd node

  M-R-Schaefer

  committed
  5673431

  Copy full SHA for 5673431

• updated jaxmd to use new callback api

  M-R-Schaefer

  committed
  ea64e13

  Copy full SHA for ea64e13

Commits on Jun 5, 2024

• deleted comment

  M-R-Schaefer

  committed
  8fd0ca1

  Copy full SHA for 8fd0ca1

• Merge branch 'dev' into shallow_ens

  M-R-Schaefer

  committed
  9f4c901

  Copy full SHA for 9f4c901

• remove old import

  M-R-Schaefer

  committed
  2150bcc

  Copy full SHA for 2150bcc

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 9b0963e
  pre-commit-ci[bot]

  committed
  9b0963e

  Copy full SHA for 9b0963e

• made shallow ensembles compatible with jaxmd

  M-R-Schaefer

  committed
  e8e0182

  Copy full SHA for e8e0182

• Merge branch 'shallow_ens' of https://github.com/apax-hub/apax into shallow_ens

  M-R-Schaefer

  committed
  ded97f6

  Copy full SHA for ded97f6

• fixed bug for using jaxmd without ensembles

  M-R-Schaefer

  committed
  cb65848

  Copy full SHA for cb65848

• account for non shallow ensemble

  M-R-Schaefer

  committed
  c7e3c58

  Copy full SHA for c7e3c58

Commits on Jun 8, 2024

• Merge branch 'dev' into shallow_ens

  M-R-Schaefer

  committed
  76a867f

  Copy full SHA for 76a867f

Commits on Jun 12, 2024

• Merge branch 'dev' into shallow_ens

  M-R-Schaefer

  committed
  8427281

  Copy full SHA for 8427281

• consistent use of std as uncertainty, not variance

  M-R-Schaefer

  committed
  6a6ad25

  Copy full SHA for 6a6ad25

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 8a86bee
  pre-commit-ci[bot]

  committed
  8a86bee

  Copy full SHA for 8a86bee

• linting

  M-R-Schaefer

  committed
  8de78f9

  Copy full SHA for 8de78f9

• Merge branch 'shallow_ens' of https://github.com/apax-hub/apax into shallow_ens

  M-R-Schaefer

  committed
  b7e2b42

  Copy full SHA for b7e2b42

Commits on Jun 20, 2024

• Merge branch 'dev' into shallow_ens

  M-R-Schaefer

  authored
  fdf859a

  Copy full SHA for fdf859a

Commits on Jul 9, 2024

• Merge branch 'dev' into shallow_ens

  M-R-Schaefer

  authored
  b284f63

  Copy full SHA for b284f63