Skip to content

Commit 36fc0d1

Browse files
fix(jax): remove jaxlib version pin to fix ARM64 compatibility
ROOT CAUSE: jaxlib==0.4.8 has no pre-built wheels for ARM64 Linux, causing all JAX-dependent tests to fail on GitHub Actions ARM64 runners. CHANGES: - Removed jaxlib==0.4.8 pin from dependencies - JAX package will automatically install compatible jaxlib version IMPACT: - Fixes CI test failures on ARM64 Linux (GitHub Actions) - JAX manages its own jaxlib dependency automatically - No functional changes to code FILES MODIFIED: - pyproject.toml Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 58c6eb1 commit 36fc0d1

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

pyproject.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,6 @@ pettingzoo = [
5151
]
5252
jax = [
5353
"jax==0.4.8",
54-
"jaxlib==0.4.8",
5554
"flax==0.6.8",
5655
"optax==0.1.4",
5756
"chex==0.1.5",

0 commit comments

Comments
 (0)