Dependency Graph for flax 0.3.3