Search results

Implement Function Transformations like Calibration for jax-md

Status: Open.

#507 In apax-hub/apax;

· M-R-Schaefer opened on Jan 20, 2026

Add NLL based post-hoc calibration

Status: Open.

#506 In apax-hub/apax;

· M-R-Schaefer opened on Jan 20, 2026

performance metrics in ApaxJaxMD might be wrong after restarts

Status: Open.

#503 In apax-hub/apax;

· PythonFZ opened on Jan 9, 2026

Migrate from Flax Linen to Flax NNX

enhancementNew feature or requestNew feature or request

Status: Open.

#500 In apax-hub/apax;

· TobiasDijkhuis opened on Jan 6, 2026

Dropout during training

enhancementNew feature or requestNew feature or request

Status: Open.

#499 In apax-hub/apax;

· TobiasDijkhuis opened on Jan 6, 2026

NPT with jaxmd only allows volume to change while keeping cell shape fixed

Status: Open.

#492 In apax-hub/apax;

· alikhamze opened on Oct 15, 2025

add <code>dump_rate</code> to ApaxBatchPrediction Node

Status: Open.

#491 In apax-hub/apax;

· PythonFZ opened on Oct 4, 2025

Training on GPU inference on CPU throws error on checkpoint loading

bugSomething isn't workingSomething isn't working

Status: Open.

#490 In apax-hub/apax;

· M-R-Schaefer opened on Oct 2, 2025

property head and <code>APAX_PROPERTIES </code>

Status: Open.

#485 In apax-hub/apax;

· PythonFZ opened on Sep 13, 2025

jax.sharding.PositionalSharding was deprecated in JAX v0.6.0 and removed in JAX v0.7.0

bugSomething isn't workingSomething isn't working

Status: Open.

#475 In apax-hub/apax;

· M-R-Schaefer opened on Jul 29, 2025

data with <code>infinite</code> stress values can not be trained on.

Status: Open.

#473 In apax-hub/apax;

· PythonFZ opened on Jul 22, 2025

Derive model path from experiment directory

Status: Open.

#449 In apax-hub/apax;

· M-R-Schaefer opened on Jun 4, 2025