Skip to content

Commit

Permalink
awesome.md (#2622)
Browse files Browse the repository at this point in the history
  • Loading branch information
abhigunj authored Nov 13, 2024
1 parent 6b79a2e commit 830b978
Show file tree
Hide file tree
Showing 3 changed files with 57 additions and 2 deletions.
9 changes: 7 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,13 @@
Official Documentation: <https://openxla.org/stablehlo>

<picture>
<source media="(prefers-color-scheme: dark)" srcset="docs/images/assets/stablehlo_ecosystem_dark.svg">
<img alt="StableHLO Ecosystem" src="docs/images/assets/stablehlo_ecosystem.svg">
<source
media="(prefers-color-scheme: dark)"
srcset="docs/images/assets/stablehlo_ecosystem_dark.svg"
class="devsite-dark-theme">
<img
alt="StableHLO Ecosystem"
src="docs/images/assets/stablehlo_ecosystem.svg">
</picture>

StableHLO is an operation set for high-level operations (HLO) in machine
Expand Down
2 changes: 2 additions & 0 deletions docs/_toc.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
48 changes: 48 additions & 0 deletions docs/awesome.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
# Awesome OpenXLA

<picture>
<source
media="(prefers-color-scheme: dark)"
srcset="https://raw.githubusercontent.com/openxla/xla/refs/heads/main/docs/images/openxla_dark.svg"
class="devsite-dark-theme">
<img
alt="OpenXLA Ecosystem"
src="https://raw.githubusercontent.com/openxla/xla/refs/heads/main/docs/images/openxla.svg">
</picture>

[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 <img align="center" src="https://img.shields.io/github/stars/jax-ml/jax?style=social">
- [PyTorch/XLA](https://github.com/pytorch/xla/) provides a bridge from PyTorch
to OpenXLA and StableHLO <img align="center" src="https://img.shields.io/github/stars/pytorch/xla?style=social">
- [TensorFlow](https://github.com/tensorflow/tensorflow) is a long-standing ML
framework with a large ecosystem <img align="center" src="https://img.shields.io/github/stars/tensorflow/tensorflow?style=social">

## 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 <img align="center" src="https://img.shields.io/github/stars/google-ai-edge/ai-edge-torch?style=social">

## Tooling and Visualization

- [Model Explorer](https://github.com/google-ai-edge/model-explorer) offers
heirarchical graph visualization with support for StableHLO models
<img align="center" src="https://img.shields.io/github/stars/google-ai-edge/model-explorer?style=social">

0 comments on commit 830b978

Please sign in to comment.