cottax.pytree.consumers.declaration module

A declared model across a jax trace: its plan, Static values and tags the key, every other value traced.

class Static(value: Any)

Bases: Tagged

A given value that sizes or selects rather than computes (a count, a switch): kept as written and part of the compiled program’s key, so a new one compiles anew. Every other value given is an array, traced.