`Autograd10YBranchLevelSet.ipynb` currently doesn't work and needs to be fixed. Biggest issue is probably aux data handling.