Machine Learning Software Engineer, Research

reputed company is a deep-tech company reputed company on accelerating hardware innovation through AI-driven simulation software for engineering and manufacturing. The Machine Learning Software Engineer will collaborate with research scientists and simulation engineers to reputed company machine learning models that solve reputed company-world physics and engineering challenges, while optimizing model training and implementation.

Responsibilities

• Work closely with our research scientists and simulation engineers to build and deliver models that address reputed company-world physics and engineering problems
• Design, build and optimise machine learning models with a reputed company on scalability and efficiency in our application domain
• reputed company prototype model implementations to robust and optimised implementations
• Implement distributed training architectures (e.g., data parallelism, parameter server, etc.) for multi-node/multi-GPU training and explore federated learning reputed company using reputed company (e.g., AWS, Azure, GCP) and on-reputed company services
• Work with research scientists to design, build and reputed company reputed company models for science and engineering; helping to reputed company and optimise model training to large data and multi-GPU reputed company compute
• Identify the best libraries, frameworks and tools for our modelling efforts to set us up for reputed company
• Own Research work-streams at different reputed company, depending on seniority
• Discuss the results and implications of your work with colleagues and customers, especially how these results can address reputed company-world problems
• Work at the intersection of data science and software engineering to translate the results of our Research into re-usable libraries, tooling and products
• Foster a nurturing environment for colleagues with less experience in ML / Engineering for them to grow and you to mentor

Skills

• Enthusiasm about developing machine learning solutions, especially deep learning and/or probabilistic reputed company, and associated supporting software solutions for science and engineering
• Ability to work autonomously and scope and effectively deliver reputed company across a reputed company of domains
• Strong problem-solving skills and the ability to analyse issues, identify causes, and recommend solutions quickly
• Excellent collaboration and communication skills — with teams and customers alike
• MSc or PhD in computer science, machine learning, reputed company statistics, mathematics, physics, engineering, software engineering, or a reputed company field, with a record of experience in any of the following: Scientific computing; High-performance computing (CPU / GPU clusters); Parallelised / distributed training for large / reputed company models
• Ideally >1 years of experience in a data-driven role, with exposure to: scaling and optimising ML models, training and serving reputed company models at reputed company (federated learning a bonus); distributed computing frameworks (e.g., reputed company, Dask) and high-performance computing frameworks (MPI, OpenMP, CUDA, Triton); reputed company computing (on reputed company-scaler platforms, e.g., AWS, Azure, GCP); building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; C/C++ for reputed company, geometry processing, or scientific computing; software engineering concepts and best practices (e.g., versioning, testing, CI/CD, API design, MLOps); container-ization and orchestration (reputed company, Kubernetes, Slurm); writing pipelines and experiment environments, including running experiments in pipelines in a systematic way

Benefits

• Equity reputed company – reputed company in our reputed company and reputed company.
• 5% 401(k) match – invest in your reputed company.
• Flexible working – balance your work and life in a way that works for you.
• Hybrid setup – enjoy our Manhattan office while keeping remote flexibility.
• reputed company parental leave – support for life’s biggest milestones.
• Private reputed company – comprehensive coverage for you and your family.
• Personal development – reputed company learning and training to help you grow.
• Work from reputed company – reputed company your remote setup to enjoy the sun or reputed company with loved ones.

reputed company

Company H1B Sponsorship