egglog rules to latex repr

[swap357]

for better rulesets readability,
use egraph.as_egglog_string to get all S-expressions,
then convert each rule(...) and rewrite(...) form into its LaTeX representation.

rendered rules and rewrite:

[swap357]

updated. latex rendering util can be used by importing it from -

from egglog_to_latex import visualize_ruleset_latex

and placing it on scripts with notebook check -

if IN_NOTEBOOK:
    # Visualize the constant propagation ruleset
    visualize_ruleset_latex(ruleset_const_propagate)

[Copilot]

Pull Request Overview

This PR adds functionality to convert Egglog rules and rewrites into LaTeX representations and integrates notebook visualization for those rulesets.

• Introduces egglog_to_latex.py with S-expression parsing, LaTeX conversion, and notebook display.
• Hooks into chapter 3 tutorial script to visualize constant-propagation and if-else folding rulesets when in a notebook.

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 3 comments.

File	Description
sealir-tutorials/egglog_to_latex.py	New module for tokenizing S-expressions, converting them to LaTeX, and visualizing in notebooks
sealir-tutorials/ch03_egraph_program_rewrites.py	Added imports and if IN_NOTEBOOK blocks to display LaTeX-rendered rulesets

Comments suppressed due to low confidence (2)

sealir-tutorials/egglog_to_latex.py:89

• Consider adding a docstring to _sexp_tex explaining its purpose and how it handles both str and list inputs.

def _sexp_tex(x) -> str:

sealir-tutorials/egglog_to_latex.py:5

• Add unit tests for tokenize (and parse_sexps) to ensure S-expression edge cases are correctly handled.

def tokenize(egglog_str: str) -> List[str]:

[Copilot]

[nitpick] Move the IPython.display import to the top of the module to avoid runtime overhead when calling visualize_ruleset_latex multiple times.

[Copilot]

If this script is intended to be run as a module, consider using a relative import (from .utils import IN_NOTEBOOK) to avoid ImportError when the package is installed.

Suggested change

	from utils import IN_NOTEBOOK
	from .utils import IN_NOTEBOOK

[Copilot]

[nitpick] These if IN_NOTEBOOK blocks are repeated—consider extracting to a helper function to DRY up the visualization calls.

Suggested change

	if IN_NOTEBOOK:
	# Visualize the constant propagation ruleset
	visualize_ruleset_latex(ruleset_const_propagate)
	def visualize_if_in_notebook(ruleset):
	if IN_NOTEBOOK:
	# Visualize the given ruleset
	visualize_ruleset_latex(ruleset)
	visualize_if_in_notebook(ruleset_const_propagate)

[sklam]

Long rewrite rules can be too long for the screen; e.g. the pade44 tanh expansion in demo1

@ruleset
def pade44_tanh_expansion(x: Term):
   ...
    yield rewrite(Npy_tanh_float32(x)).to(
        div(
            add(mul(flt(10), pow(x, liti64(3))), mul(flt(105), x)),
            add(
                add(pow(x, liti64(4)), mul(flt(45), pow(x, liti64(2)))),
                flt(105),
            ),
        )
    )

The denominator gets very long.

Needs a way to split the long equations.

[sklam]

When converted to HTML pages, somehow the \ is visible.

In live notebook, they are currently rendered (hidden).

[sklam]

I used a modified version of this with ability to replace long names for shorter ones. Compare with https://github.com/numba/numba-prototypes/blob/archive/scipy_2025_slides/sealir-tutorials/utils/egglog_to_latex.py