Enhanced AI tracks neurons in moving animals