v0.25.0: Increase Jax version range (using tf-nightly)

What's Changed

• Increase Jax version range and use tf-nightly by @araffin in #84

Full Changelog: v0.24.0...v0.25.0