Dependency Graph for flax 0.3.1