Visualizing HLO computation graphs

visu-hlo is a Python package that displays the HLO (High Level Operations) representation of JAX functions as SVG visualizations. This tool helps developers understand the computational graphs generated by XLA compilations.

Features

🎯 Easy Visualization: Display HLO graphs with a single function call
⚡ JIT Support: Works with both regular and jitted JAX functions
🖼️ SVG Output: High-quality vector graphics that scale perfectly
🖥️ Cross-Platform: Supports Linux, macOS, and Windows
📦 Lightweight: Minimal dependencies - just JAX and Graphviz

Quick Example

import jax.numpy as jnp
from visu_hlo import show

# Display optimized HLO (default)
show(lambda x: 3 * x * 2, jnp.ones(10))

To display the non-optimized HLO:

show(lambda x: 3 * x * 2, jnp.ones(10), jit=False)

To save as an SVG file:

from visu_hlo import write_svg
write_svg('graph.svg', func, jnp.ones(10))

Installation

pip install visu-hlo

System dependency: Install Graphviz

Documentation

Full documentation: https://visu-hlo.readthedocs.io/

License

MIT

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

visu_hlo-1.1.0.tar.gz (28.0 kB view details)

Uploaded Apr 1, 2026 Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

visu_hlo-1.1.0-py3-none-any.whl (7.1 kB view details)

Uploaded Apr 1, 2026 Python 3

File details

Details for the file visu_hlo-1.1.0.tar.gz.

File metadata

Download URL: visu_hlo-1.1.0.tar.gz
Upload date: Apr 1, 2026
Size: 28.0 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for visu_hlo-1.1.0.tar.gz
Algorithm	Hash digest
SHA256	`92aa3d256047dc115409deb5914bd923f1d162a5a8b9746e4085bd8ae4844d79`
MD5	`f60e9970fce7138041788c15aaf889ef`
BLAKE2b-256	`89a031006311e329d791b61b1ce41f8f366491eee20404351019c29cf2832468`

See more details on using hashes here.

Provenance

The following attestation bundles were made for visu_hlo-1.1.0.tar.gz:

Publisher: release.yml on CMBSciPol/visu-hlo

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Statement:
- Statement type: https://in-toto.io/Statement/v1
- Predicate type: https://docs.pypi.org/attestations/publish/v1
- Subject name: visu_hlo-1.1.0.tar.gz
- Subject digest: 92aa3d256047dc115409deb5914bd923f1d162a5a8b9746e4085bd8ae4844d79
- Sigstore transparency entry: 1205630367
- Sigstore integration time: Apr 1, 2026
Source repository:
- Permalink: CMBSciPol/visu-hlo@0515e1dff3fd46d7f7e1133fe84fbe4c46fb5132
- Branch / Tag: refs/tags/v1.1.0
- Owner: https://github.com/CMBSciPol
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: release.yml@0515e1dff3fd46d7f7e1133fe84fbe4c46fb5132
- Trigger Event: release