/xla2onnx

Primary LanguagePython

xla2onnx

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