flax
add set_graph_mode
#5269
Merged

Loading