New features
- Added handlers for the new C-based JAX pytree classes (jax-ml/jax#24906), which are no longer dataclasses. The new handlers preserve the original dataclass-like rendering behavior and support upgraded JAX versions.
Other changes
- Updated
uv
lockfile to use newer package versions when setting up the development environment and running tests. - Silenced some warnings that were triggering in tests.