Repository navigation

Preserve initialized loss dependencies and avoid cached JAX tracers - #183

Open

rogeriojorge wants to merge 4 commits into

mainfrom

fix/loss-dependencies

Open

Preserve initialized loss dependencies and avoid cached JAX tracers#183
rogeriojorge wants to merge 4 commits into
mainfrom
fix/loss-dependencies

Commits on Oct 23, 2026
- Preserve loss dependencies and avoid cached JAX tracers
  rogeriojorge
  committed
  2f3a353
- Cover composite initialization and shared dependency gradients
  rogeriojorge
  committed
  fd2bb49
Commits on Oct 26, 2026
- Merge remote-tracking branch 'origin/main' into m183
  rogeriojorge
  committed
  b22e3a5
Commits on Oct 31, 2026
- Merge remote-tracking branch 'origin/main' into fix/loss-dependencies
  rogeriojorge
  committed
  937e363