Dependency Graph for flax 0.4.0