Visualize, create, and operate on JAX PyTree in the most intuitive way possible
pytreeclass is a JAX-compatible class builder to create and operate on stateful JAX PyTrees in a performant and intuitive way, by building on familiar concepts found in numpy, dataclasses, and others.