flax

c9cebeee - support passing arguments directly to the struct.dataclass decorator