Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Initialize JAX configuration as per the v0.4.25 release #92

Merged
merged 8 commits into from
Feb 28, 2024
Merged

Initialize JAX configuration as per the v0.4.25 release #92

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
c2d970b
Initialize JAX configuration as per the release
flferretti Feb 27, 2024
68ab8a2
Remove --forked from pytest as workaround for jax 0.4.25
traversaro Feb 27, 2024
6eaf105
Merge pull request #94 from ami-iit/fixnewjaxcrash
flferretti Feb 27, 2024
48c8396
Constrain jax to be lower of 0.4.25 until we switch to functional
traversaro Feb 27, 2024
032dee5
Add constraint also to jaxlib
traversaro Feb 27, 2024
9fe432f
Merge branch 'main' into flferretti-patch-1
flferretti Feb 28, 2024
10d4677
Revert "Remove --forked from pytest as workaround for jax 0.4.25"
flferretti Feb 28, 2024
b5b4f31
Merge pull request #95 from ami-iit/revert-94-fixnewjaxcrash
flferretti Feb 28, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/jaxsim/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,12 @@
def _jnp_options() -> None:
import os

from jax.config import config
import jax

# Enable by default
if not ("JAX_ENABLE_X64" in os.environ and os.environ["JAX_ENABLE_X64"] == "0"):
logging.info("Enabling JAX to use 64bit precision")
config.update("jax_enable_x64", True)
jax.config.update("jax_enable_x64", True)

import jax.numpy as jnp
import numpy as np
Expand Down
Loading