Machine Learning Software Engineer, Libraries
19 hours ago
PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software.
We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries. By enabling high-fidelity, multi-physics simulation through AI inference across the entire engineering lifecycle, PhysicsX unlocks new levels of optimization and automation in design, manufacturing, and operations - empowering engineers to push the boundaries of possibility. Our customers include leading innovators in Aerospace & Defense, Materials, Energy, Semiconductors, and Automotive.
The Role
PhysicsX is developing a platform used by Data Scientists and Simulation Engineers to build, train and deploy Deep Physics Models. The core of this platform is powered by several Machine Learning Python Libraries handling everything from data processing, to deep learning model training, and complex optimisation algorithms.
The Libraries team is responsible for designing, developing, and maintaining those Machine Learning Libraries. As a Machine Learning Software Engineer within the team, you will be the crucial link between cutting-edge research and production-ready software. You will collaborate closely with our Research team to translate the latest state-of-the-art models and techniques into robust, easy-to-use, and performant library code. You are a software engineer who thinks like a data scientist, ensuring our libraries are not only robust but also embody the best practices of the machine learning lifecycle.
Note: We are currently recruiting for multiple positions, however please only apply for the role that best aligns with your skillset and career goals.
What you will do
- Collaborate with product managers and library users to translate functional requirements into robust technical specifications and production-ready code.
- Work closely with our research scientists to productionize novel models and algorithms, integrating them into our core libraries.
- Contribute high-quality, tested, documented, and maintainable code across the library stack, ensuring our core systems are reliable and easy for other engineers to build upon.
- Champion software engineering excellence across the team and company, driving best practices in testing (unit, integration, performance), CI/CD, and system architecture.
- Own your work end-to-end, from ideation and experimentation through to deployment and maintenance in a small, agile team environment.
- Continuously explore and evaluate new techniques and frameworks, helping to guide the technical evolution of our platform.
- Share your knowledge with the team, contribute to our engineering guild, and stay connected with the broader community.
- Professional experience with C++ and/or GPGPU programming (e.g., CUDA, Triton) for performance optimization.
- Experience building and maintaining widely-used internal or open-source libraries.
- Experience applying machine learning to scientific domains, particularly with 3D geometric or simulation data.
- Familiarity with model packaging and serialization techniques (e.g., ONNX, TorchScript).
- A passion for the craft of software engineering and building robust, maintainable systems in a collaborative, cross-functional team.
- Solid software engineering skills, with a focus on writing clean, testable, and maintainable code.
- Strong problem-solving skills, with the ability to diagnose complex issues in a high-uncertainty R&D environment.
- A commitment to owning the software development lifecycle, including testing, containerization, and CI/CD.
- A deep understanding of machine learning fundamentals (e.g., model architectures, optimizers, statistical principles) and the data science lifecycle.
- Strong proficiency in Python's scientific computing and Machine Learning ecosystem (e.g., PyTorch, NumPy, SciPy, Pandas).
- An understanding of how high-level Machine Learning frameworks interact with low-level hardware (e.g., C++/CUDA on GPUs).
- A proactive ownership mindset and the ability to navigate ambiguity.
- Excellent collaboration and communication skills, with an ability to discuss complex technical concepts with both researchers and product teams.
- Equity options - share in our success and growth.
- 5% 401(k) match - invest in your future.
- Flexible working - balance your work and life in a way that works for you.
- Hybrid setup - enjoy our Manhattan office while keeping remote flexibility.
- Enhanced parental leave - support for life's biggest milestones.
- Private healthcare - comprehensive coverage for you and your family.
- Personal development - access learning and training to help you grow.
- Work from anywhere - extend your remote setup to enjoy the sun or reconnect with loved ones.
Salary range:
$120,000 - 240,000 depending on experience
Seniority will be assessed throughout our interview process
We value diversity and are committed to equal employment opportunity regardless of sex, race, religion, ethnicity, nationality, disability, age, sexual orientation or gender identity. We strongly encourage individuals from groups traditionally underrepresented in tech to apply. To help make a change, we sponsor bright women from disadvantaged backgrounds through their university degrees in science and mathematics.
We collect diversity and inclusion data solely for the purpose of monitoring the effectiveness of our equal opportunities policies and ensuring compliance with UK employment and equality legislation. This information is confidential, used only in aggregate form, and will not influence the outcome of your application.
-
Software Engineer, Libraries
18 hours ago
New York, NY, United States PhysicsX Full timeAbout us PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software. We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries. By enabling high-fidelity, multi-physics simulation through AI inference across...
-
Software Engineer, Libraries
4 days ago
New York, NY, United States PhysicsX Full timeAbout us PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software. We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries. By enabling high-fidelity, multi-physics simulation through AI inference across...
-
Software Engineer, Libraries
7 hours ago
New York, NY, United States PhysicsX Full timeAbout us PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software. We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries. By enabling high-fidelity, multi-physics simulation through AI inference across...
-
Staff Software Engineer
5 days ago
New York, NY, United States Apple Full timeWeekly Hours: 40 Role Number: 200618202-2459 Summary At Apple, we work every day to create products that enrich people’s lives. Apple Ads makes it possible for people around the world to easily access informative and imaginative content on their devices while helping publishers and developers promote and monetize their work. Today, our technology and...
-
Machine Learning Engineer
2 days ago
New York, NY, United States Jane Street Full timeAbout the Program Our Machine Learning Engineering Internship is designed to give you a real sense of what it's like to work at Jane Street full-time as a Machine Learning Engineer. Over the course of your program, you will explore ways to approach and solve cutting-edge ML problems through fun and challenging classes, interactive sessions, and group...
-
Machine Learning Engineer
1 week ago
New York, NY, United States Jane Street Full timeAbout the Program Our Machine Learning Engineering Internship is designed to give you a real sense of what it's like to work at Jane Street full-time as a Machine Learning Engineer. Over the course of your program, you will explore ways to approach and solve cutting-edge ML problems through fun and challenging classes, interactive sessions, and group...
-
Staff Software Engineer, Machine Learning
5 days ago
New York, NY, United States Warner Bros. Discovery Full timeWelcome to Warner Bros. Discovery... the stuff dreams are made of. Who We Are... When we say, "the stuff dreams are made of," we're not just referring to the world of wizards, dragons and superheroes, or even to the wonders of Planet Earth. Behind WBD's vast portfolio of iconic content and beloved brands, are the storytellers bringing our characters to life,...
-
Software Engineer, Machine Learning
2 days ago
New York, NY, United States Figma Full timeFigma is growing our team of passionate creatives and builders on a mission to make design accessible to all. Figma's platform helps teams bring ideas to life-whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI. From idea to product, Figma empowers teams to streamline workflows, move faster, and work...
-
Software Engineer, Machine Learning
1 week ago
New York, NY, United States Figma Full timeFigma is growing our team of passionate creatives and builders on a mission to make design accessible to all. Figma's platform helps teams bring ideas to life-whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI. From idea to product, Figma empowers teams to streamline workflows, move faster, and work...
-
Software Engineer, Machine Learning
4 days ago
New York, NY, United States Figma Full timeFigma is growing our team of passionate creatives and builders on a mission to make design accessible to all. Figma's platform helps teams bring ideas to life-whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI. From idea to product, Figma empowers teams to streamline workflows, move faster, and work...
