Selam Waktola issues

Repositories
Issues
Comments

Results 4 issues of


                                            Selam Waktola

Added code examples for ix_ function with better doc string.

DOC: Improved documentation for numpy.frombuffer

Part of https://github.com/google/jax/issues/21461

pull ready

Fix broken links on README.md

Fix citation for Chex.

Remove TensorFlow dependency and migrate cifar10_resnet to Flax NNX

This PR addresses issue #[656](https://github.com/google-deepmind/optax/issues/656) by removing the TensorFlow dependency from the `cifar10_resnet` notebook. Additionally, it updates the notebook to use **Flax NNX** instead of `flax.linen`, aligning the example with...