Machine Learning Software Engineer, Research - Remote

Overview

PhysicsX is a deep-tech company focused on accelerating hardware innovation through AI-driven simulation software for engineering and manufacturing across advanced industries.

In Short

• Collaborate with research scientists and engineers to build models for real-world physics problems.
• Design and optimize machine learning models for scalability and efficiency.
• Transform prototype models into robust implementations.
• Implement distributed training architectures for multi-node/multi-GPU training.
• Scale and optimize model training for large datasets and multi-GPU cloud compute.
• Identify libraries and tools for modeling efforts.
• Own research work-streams based on seniority.
• Discuss results and implications of research with colleagues and customers.
• Translate research results into reusable libraries and products.
• Mentor colleagues in ML and engineering.

Requirements

• MSc or PhD in a relevant field (computer science, machine learning, etc.).
• Experience in scientific computing and high-performance computing.
• 1+ years in a data-driven role with exposure to ML model scaling.
• Familiarity with distributed computing frameworks.
• Experience with cloud computing on platforms like AWS, Azure, GCP.
• Proficient in Python and common ML libraries.
• Knowledge of C/C++ for scientific computing.
• Understanding of software engineering best practices.
• Experience with containerization and orchestration tools.
• Ability to write systematic experiment pipelines.

Benefits

• Make an impact in an early-stage company.
• Work with a supportive and challenging team.
• Experience a flat hierarchy encouraging idea sharing.
• Work-life balance with flexible working options.
• Competitive compensation and equity package.
• Generous vacation and parental leave.
• Access to first-class equipment for remote work.