thrml

What problem does it solve?

This Skill streamlines the creation and simulation of THRML code, enabling users to efficiently build and manipulate JAX-based probabilistic models for complex sampling and training tasks.

Core Features & Use Cases

• THRML Code Generation: Automatically generate THRML code for Ising machines and Boltzmann machines, tailored to specific sampling and training needs.
• Simulation: Simulate models using block Gibbs sampling and training for probabilistic graphical models.
• Use Case: Suppose you are developing a Boltzmann machine for image processing. Utilize this Skill to create the model architecture, specify the sampling schedule, and generate initial parameters for your model.

Quick Start

Run the 'thrml' skill with the command: /thrml create IsingMachine --nodes 100 --edges 200 --biases normal(0, 1) --weights normal(0, 1) --beta 1.0