PhysicsX · 2 hours ago
Machine Learning Software Engineer, Libraries
Manhattan, NY
Full-time
Hybrid
Mid, Senior Level
$120K/yr - $240K/yrPhysicsX is a deep-tech company focused on accelerating hardware innovation through AI-driven simulation software. The Machine Learning Software Engineer will collaborate with product managers and research scientists to develop and maintain machine learning libraries, ensuring robust and production-ready code that meets functional requirements.
Artificial Intelligence (AI)Information TechnologyMachine LearningSemiconductorSimulationSoftware
Responsibilities
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
Qualification
Required
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
Benefits
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.
Company
PhysicsX
PhysicsX offers an AI-native simulation software stack for engineering and manufacturing across advanced industries.
Founded in 2019
51-200 employees
H1B Sponsorship
PhysicsX has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2024 (1)
Funding
Current Stage
Growth StageTotal Funding
$187MKey Investors
NVenturesAtomicoGeneral Catalyst
2025-11-19Series B· $20M
2025-06-22Series B· $135M
2023-11-27Series A· $32M
Leadership Team
Jacomo Corbo
CEO & Co-Founder
Recent News
Tech Funding News
2025-11-08
Company data provided by crunchbase