g2n 0.4.2

A compiler backend for PyTorch: pointwise fusion, buffer-reuse planning, and a persistent cross-run compile cache.

g2n

A compiler backend for PyTorch: pointwise fusion, buffer-reuse planning, and a persistent cross-run compile cache so repeat builds skip recompilation.

pip install g2n

import torch
import g2n

model = MyModule().eval()
compiled = g2n.compile(model)
# or register as a torch.compile backend:
compiled = torch.compile(model, backend="g2n")

A license key unlocks the Pro and Enterprise features (enhanced buffer planner, persistent cache, multi-accelerator routing, model-zoo configurations). See the documentation at https://g2n.dev/docs.