What's Changed
- Drop support for python 3.10, add support for python 3.12 #252 #253
- Update JAX nightly index usage #238
- Add new attr use_shardy_partitioner in _xla_call_module attr checking. #240
- Update tests to use Flax instead of haiku #241
- Add support for fresnel_sin and fresnel_cos #243
- Use V2 sharding if available. #244
- Fix TensorListGetItem tests #251
- JAX api migration and tests updates. #231 #232 #233 #235 #236 #237 #239 #245 #246 #248 #249 #250
Full Changelog: v0.3.7...v0.3.8