Upgrade the SPU library to version 0.9.4 in this repository.
SPU v0.9.4 introduces a major version upgrade to its JAX dependency. This requires updating the JAX package to the version specified by SPU 0.9.4 and ensuring compatibility across the codebase.
Action Items:
- Update SPU to v0.9.4 in project dependencies.
- Upgrade JAX to the required major version.
- Validate code compatibility with the new JAX version and resolve any breaking changes.
- Run the full test suite to verify stability and correctness after upgrade.
- Update documentation and dependency specifications as needed.
Upgrade the SPU library to version 0.9.4 in this repository.
SPU v0.9.4 introduces a major version upgrade to its JAX dependency. This requires updating the JAX package to the version specified by SPU 0.9.4 and ensuring compatibility across the codebase.
Action Items: