Convert XLA to ONNX.
./run.sh
to run all test../format.sh
to run format.
TODO
- MNIST (JAX)
- Resnet (JAX)
- grad(MNIST) (JAX)
- grad(Resnet) (JAX)
- MNIST (TensorFlow)
- Resnet50 (TensorFlow)
- Python package
- brax
- General reduction support (not using ReduceSum/ReduceMax)
- license