Giters

take-cheeze / xla2onnx

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Languages

Language:Python 98.4%Language:Shell 1.6%