Software Engineer | ML infra researcher