Code

feat(jax): force & virial#4251

Code

Merged

iProzd merged 8 commits intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom njzjz:jax-forcenjzjz/deepmd-kit:jax-forceCopy head branch name to clipboardOct 29, 2024

+284-17

Commits

Commits on Oct 25, 2024

• feat(jax): force

  Show description for 0517b59
  njzjz

  committed
  0517b59

  Copy full SHA for 0517b59

• do not change diff directly

  Show description for d0b576f
  njzjz

  authored
  d0b576f

  Copy full SHA for d0b576f

• support virial

  Show description for 004b89a
  njzjz

  committed
  004b89a

  Copy full SHA for 004b89a

• resolve comments

  Show description for b9eefd3
  njzjz

  committed
  b9eefd3

  Copy full SHA for b9eefd3

• remove devices

  Show description for 1bbb1c2
  njzjz

  committed
  1bbb1c2

  Copy full SHA for 1bbb1c2

• do_atomic_virial cannot jit

  Show description for 23e318a
  njzjz

  authored
  23e318a

  Copy full SHA for 23e318a

Commits on Oct 26, 2024

• fix bug before following refactor

  Show description for c34d8b4
  njzjz

  committed
  c34d8b4

  Copy full SHA for c34d8b4

• refactor

  Show description for 19e0970
  njzjz

  committed
  19e0970

  Copy full SHA for 19e0970