Admins will be upgrading ORNL GitLab Servers on Saturday, 16 May 2026, from 7 AM until 11 AM EST. Repositories will experience intermittent outages during this time.
# Chains will be drawn sequentially. If you are running MCMC in CPU, consider using `numpyro.set_host_device_count(2)` at the beginning of your program.
# You can double-check how many devices are available in your system using `jax.local_device_count()`.
"-Wignore::UserWarning"
# FutureWarning: In the future `np.object` will be defined as the corresponding NumPy scalar.
"-Wignore::FutureWarning"
];
disabledTests=[
# ValueError: Found unexpected Arrays on value of type <class 'list'> in static attribute 'layers'
# of Pytree type '<class 'test_module.test_random_nnx_module_mcmc_sequence_params.<locals>.MLP'>'.
# This is an error starting from Flax version 0.12.0.