Dependency Graph for flax 0.5.3