flax
permit passing mesh or rules to spmd primitives
#2939
Merged

Loading