KLay Docs

KLay is a Python library for accelerating inference on sparse arithmetic circuits.

Get started using the Quick Start Guide!

At its core, KLay transforms sparse directed acyclic graphs into layers that can be executed in parallel. The design of KLay is described in our paper KLay: Accelerating Arithmetic Circuits for Neurosymbolic AI, published at ICLR 2025.

---

Contents

• Home
• Quick Start Guide
  • Installation
  • Usage
• Circuit Creation Tutorial
  • Loading Circuits
  • Multi-Rooted Circuits
  • Manual Circuits
• Circuit Evaluation Tutorial
  • Backends
  • Inference
• API Reference
  • Circuit
    • Circuit.nb_nodes()
    • Circuit.nb_root_nodes()
    • Circuit.true_node()
    • Circuit.false_node()
    • Circuit.literal_node()
    • Circuit.and_node()
    • Circuit.or_node()
    • Circuit.set_root()
    • Circuit.remove_unused_nodes()
    • Circuit.add_sdd()
    • Circuit.add_sdd_from_file()
    • Circuit.add_d4_from_file()
    • Circuit.to_torch_module()
    • Circuit.to_jax_function()