Skip to content

feat: JAX training - #4782

Closed
njzjz wants to merge 40 commits into
deepmodeling:masterfrom
njzjz:jax_training_submit
Closed

feat: JAX training#4782
njzjz wants to merge 40 commits into
deepmodeling:masterfrom
njzjz:jax_training_submit

Commits

Commits on May 25, 2025

  • checkpoint

    committed

Commits on May 26, 2025

Commits on May 27, 2025

Commits on May 28, 2025

Commits on Jun 1, 2025

Commits on Jun 3, 2025

  • restart. bug to be fixed

    committed
  • fix lr

    committed

Commits on Jun 10, 2025

Commits on Jun 17, 2025

Commits on Jun 20, 2025

  • fix nopbc behavior

    committed
  • should be >1

    committed

Commits on Jun 25, 2025

Commits on Jul 10, 2025

  • pass mixed_type

    committed

Commits on Jul 27, 2025

Commits on Jul 28, 2025

Commits on Nov 27, 2025

Commits on Dec 2, 2025