Back to jobs

Machine Learning Software Engineer, Libraries

New York, United States

About 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 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.

 

Ideally

  • 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).

 

What you bring to the table

  • 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.

 

What we offer

  • 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. 
 

Apply for this job

*

indicates a required field

Resume/CV*

Accepted file types: pdf, doc, docx, txt, rtf

Cover Letter

Accepted file types: pdf, doc, docx, txt, rtf

Select...
Select...

Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in PhysicsX’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Select...
Select...
Race & Ethnicity Definitions

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

Select...

Voluntary Self-Identification of Disability

Form CC-305
Page 1 of 1
OMB Control Number 1250-0005
Expires 04/30/2026

Why are you being asked to complete this form?

We are a federal contractor or subcontractor. The law requires us to provide equal employment opportunity to qualified people with disabilities. We have a goal of having at least 7% of our workers as people with disabilities. The law says we must measure our progress towards this goal. To do this, we must ask applicants and employees if they have a disability or have ever had one. People can become disabled, so we need to ask this question at least every five years.

Completing this form is voluntary, and we hope that you will choose to do so. Your answer is confidential. No one who makes hiring decisions will see it. Your decision to complete the form and your answer will not harm you in any way. If you want to learn more about the law or this form, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

How do you know if you have a disability?

A disability is a condition that substantially limits one or more of your “major life activities.” If you have or have ever had such a condition, you are a person with a disability. Disabilities include, but are not limited to:

  • Alcohol or other substance use disorder (not currently using drugs illegally)
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, HIV/AIDS
  • Blind or low vision
  • Cancer (past or present)
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or serious difficulty hearing
  • Diabetes
  • Disfigurement, for example, disfigurement caused by burns, wounds, accidents, or congenital disorders
  • Epilepsy or other seizure disorder
  • Gastrointestinal disorders, for example, Crohn's Disease, irritable bowel syndrome
  • Intellectual or developmental disability
  • Mental health conditions, for example, depression, bipolar disorder, anxiety disorder, schizophrenia, PTSD
  • Missing limbs or partially missing limbs
  • Mobility impairment, benefiting from the use of a wheelchair, scooter, walker, leg brace(s) and/or other supports
  • Nervous system condition, for example, migraine headaches, Parkinson’s disease, multiple sclerosis (MS)
  • Neurodivergence, for example, attention-deficit/hyperactivity disorder (ADHD), autism spectrum disorder, dyslexia, dyspraxia, other learning disabilities
  • Partial or complete paralysis (any cause)
  • Pulmonary or respiratory conditions, for example, tuberculosis, asthma, emphysema
  • Short stature (dwarfism)
  • Traumatic brain injury
Select...

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.