Code

ML with separate gradient steps#587

Code

Draft

jfowkes wants to merge 12 commits intodevptycho/ptypy:devfrom ML-separate-gradsptycho/ptypy:ML-separate-gradsCopy head branch name to clipboard

+4,338-10

Commits

Commits on Jan 6, 2025

• ML with separate gradient steps

  jfowkes

  committed
  af75589

  Copy full SHA for af75589

• Add all poly coeffs to ML_separate_grads

  jfowkes

  committed
  b653600

  Copy full SHA for b653600

• Add Poisson and Euclid models to ML_separate_grads

  jfowkes

  committed
  ef7e14e

  Copy full SHA for ef7e14e

• Serialize ML with separate gradient steps

  jfowkes

  committed
  ae6f01b

  Copy full SHA for ae6f01b

• Drop probe_update_start from MLSeparateGrads

  jfowkes

  committed
  17ba4ad

  Copy full SHA for 17ba4ad

• Add required kernel function

  jfowkes

  committed
  fe0df7d

  Copy full SHA for fe0df7d

• Add kernel regression test

  jfowkes

  committed
  efe000f

  Copy full SHA for efe000f

• Fix kernel regression test

  jfowkes

  committed
  60c7e60

  Copy full SHA for 60c7e60

Commits on Jan 7, 2025

• Add CUDA kernel

  jfowkes

  authored
  7c1a288

  Copy full SHA for 7c1a288

Commits on Jan 13, 2025

• Add cupy and pycuda kernels

  jfowkes

  committed
  9ad9985

  Copy full SHA for 9ad9985

• Add cupy and pycuda kernel regression tests

  jfowkes

  committed
  b25b4f6

  Copy full SHA for b25b4f6

• CuPy and PyCUDA MLSeparateGrads

  jfowkes

  committed
  1a24467

  Copy full SHA for 1a24467