What's Changed
- optimize transpose(convolution) -> convolution by @Pangoraw in #117
- Force bump xla commit by @wsmoses in #119
- Simplify path infra by @wsmoses in #128
- Bump internals by @wsmoses in #130
- Attempt gpu ci fix by @wsmoses in #125
- Fast path slice contiguous constant by @wsmoses in #137
- Add jaxmd tests by @wsmoses in #136
- Transpose batch by @wsmoses in #138
- Maxtext by @wsmoses in #139
- implement derivative for real and imag by @Pangoraw in #145
- Update compile_with_xla.cc by @klucke in #147
- Clamp derivatives by @Pangoraw in #148
- IfOp by @Pangoraw in #149
- bump enzyme and jax version by @Pangoraw in #150
- Update test-requirements.txt by @wsmoses in #155
- Try a100 CI by @wsmoses in #156
- CUDA ci by @wsmoses in #154
- add neuralgcm by @wsmoses in #141
- Generic batch op interface by @Pangoraw in #151
- Use absl for neuralgcm by @wsmoses in #158
- DynamicUpdateSliceOp reverse derivative by @Pangoraw in #159
- Try caching again by @wsmoses in #163
- WhileOp reverse derivative by @Pangoraw in #160
- remove transform.with_named_sequence attribute by @Pangoraw in #153
- Keras3 by @wsmoses in #166
- Merge concatenate of contiguous slices in a single slice by @Pangoraw in #168
- Improve enzyme gradient ops removal in while op by @Pangoraw in #167
- Fix stablehlo ffi by @wsmoses in #171
- Nicer hlo ffi by @wsmoses in #172
New Contributors
Full Changelog: v0.0.8...v0.0.9
Contributors
wsmoses, Pangoraw, and klucke