From 830b9787c58b400f380c2f945bd00a0675307cc3 Mon Sep 17 00:00:00 2001 From: Abhinav Gunjal Date: Tue, 12 Nov 2024 20:14:52 -0800 Subject: [PATCH] awesome.md (#2622) --- README.md | 9 +++++++-- docs/_toc.yaml | 2 ++ docs/awesome.md | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 57 insertions(+), 2 deletions(-) create mode 100644 docs/awesome.md diff --git a/README.md b/README.md index 349c0ecc35..b1fad7217f 100644 --- a/README.md +++ b/README.md @@ -3,8 +3,13 @@ Official Documentation: - - StableHLO Ecosystem + + StableHLO Ecosystem StableHLO is an operation set for high-level operations (HLO) in machine diff --git a/docs/_toc.yaml b/docs/_toc.yaml index edb9dd4524..b0d7ff765a 100644 --- a/docs/_toc.yaml +++ b/docs/_toc.yaml @@ -27,6 +27,8 @@ toc: path: /stablehlo/reference - title: Roadmap path: /stablehlo/roadmap + - title: Awesome OpenXLA + path: /stablehlo/awesome - title: Tutorials section: - title: JAX Export to StableHLO diff --git a/docs/awesome.md b/docs/awesome.md new file mode 100644 index 0000000000..3ebe29cbd8 --- /dev/null +++ b/docs/awesome.md @@ -0,0 +1,48 @@ +# Awesome OpenXLA + + + + OpenXLA Ecosystem + + +[OpenXLA](https://openxla.org) is open ecosystem of performant, portable, and +extensible machine learning (ML) infrastructure components that simplify ML +development by defragmenting the tools between frontend frameworks and hardware +backends. Built by industry leaders in AI modeling, software, and hardware. + +**How is the community using OpenXLA?** This page consolidates links to +repositories and projects using OpenXLA to provide inspiration and code pointers! + +**Have a project that uses OpenXLA?** Send us a pull request and add it to this page! + +## Frameworks + +- [JAX](https://github.com/jax-ml/jax) is a ML framework with a +NumPy-like API for writing high-performance ML models +- [PyTorch/XLA](https://github.com/pytorch/xla/) provides a bridge from PyTorch +to OpenXLA and StableHLO +- [TensorFlow](https://github.com/tensorflow/tensorflow) is a long-standing ML +framework with a large ecosystem + +## PJRT Plugins + +- [libTPU](https://cloud.google.com/tpu/docs/runtimes) allows models to execute +on Google's Cloud TPUs + +## Edge Compilation + +- [Google AI Edge](https://ai.google.dev/edge) uses StableHLO as an input format +to deploy to mobile devices using [LiteRT](https://ai.google.dev/edge/litert) + - [AI Edge Torch](https://github.com/google-ai-edge/ai-edge-torch) exports + PyTorch models for mobile deployment via StableHLO + +## Tooling and Visualization + +- [Model Explorer](https://github.com/google-ai-edge/model-explorer) offers +heirarchical graph visualization with support for StableHLO models +