Metamorphic Additive Manufacturing Ltd Logo

Metamorphic Additive Manufacturing Ltd

ML Research Engineer (Performance Engineering)

Posted 8 Hours Ago
Be an Early Applicant
In-Office
Palo Alto, CA, USA
200K-280K Annually
Mid level
In-Office
Palo Alto, CA, USA
200K-280K Annually
Mid level
Optimize training and inference performance of foundation models by writing and tuning CUDA/Triton GPU kernels, profiling and removing bottlenecks, implementing low-precision and mixed-precision strategies, optimizing MoE routing and expert dispatch, integrating high-performance libraries, and collaborating closely with researchers to translate model ideas into scalable, efficient implementations.
The summary above was generated by AI
About Metamorphic

Metamorphic is developing new approaches to intelligence by combining machine learning with large-scale experimental neuroscience, informed by the principles that make the brain efficient, flexible, and robust. We are building foundation models trained on rich, continuous neural data — a high-resolution model of the brain at a scale never before possible.

Our founding team spans machine learning, neuroscience, and neurotechnology, with prior work including the MICrONS project, Neuropixels, and the Enigma project, as well as foundational scientific contributions in learning, neural computation, and generative modeling. Our work sits at the frontier of AI research, and we believe the highest-impact discoveries will come from researchers and engineers working as a single, tightly collaborative team.

The name Metamorphic reflects our belief that the next advances in intelligence will come from a change in form, beyond scale — from artificial to natural intelligence.

About the Role

We are seeking Research Engineers to join our growing AI research team. You will be responsible for maximizing the training and inference performance of Metamorphic's foundation models, from quantization and low-precision training, to MoE routing optimization, to writing custom CUDA/Triton kernels for our novel architecture. This is a high-impact, technically deep role at the frontier of ML research and engineering. You will write and optimize GPU kernels, profile and eliminate performance bottlenecks, tune low-precision training strategies, and work closely with researchers to ensure architectural decisions translate to efficient and scalable implementations. You'll have substantial autonomy to shape foundational technical decisions on a small, high-impact team.

You'll thrive in this role if you:

  • Have significant software engineering experience and can move quickly without sacrificing rigor

  • Are able to balance research goals with practical engineering constraints

  • Are able to turn theory and practice, translating paper ideas into robust and performant implementations.

  • Get excited about the nitty gritty engineering details and incremental performance improvements that others gloss over.

  • Are happy to take on tasks outside your job description to support the team

  • Enjoy pair programming and deeply collaborative work

  • Are eager to learn more about machine learning research in a novel scientific domain

  • Are enthusiastic to work at an organization that functions as a single, cohesive team pursuing large-scale AI research

  • Have ambitious goals for AI progress and are excited to create the best outcomes over the long term

We offer:

  • The chance to work on one of the most scientifically consequential AI projects being pursued today

  • A small, world-class team where your contributions directly shape the science and the company

  • Competitive compensation and benefits, along with visa sponsorship

  • Strong mentorship and career development

Salary Range

$200,000 - $280,000 USD

Based on experience. We additionally offer a competitive equity package and comprehensive benefits, as well as visa sponsorship for international candidates.

Minimum Qualifications
  • Bachelor's degree or higher in Computer Science, Machine Learning, or a related field

  • Strong software engineering skills with a proven track record of building complex systems

  • Strong proficiency in CUDA, Triton, or similar, with demonstrated experience writing and optimizing GPU kernels

  • Hands-on experience with mixed-precision and low-precision training and a practical understanding of numerical stability tradeoffs

  • Deep knowledge of transformer architectures at the implementation level

  • Experience with MoE architectures: routing algorithms, load balancing, and the systems-level challenges of expert dispatch across GPUs

  • Hands-on experience with GPU profiling tools (Nsight Compute, Nsight Systems, PyTorch Profiler)

  • Experience integrating, customizing, and extending third-party high-performance libraries (FlashAttention, cuDNN, Triton, Quack, or similar) into production training stacks

Nice to Have
  • Experience with CUTLASS, cuDNN APIs, and NCCL internals

  • Familiarity with inference optimization techniques and serving frameworks

  • Familiarity with diffusion models or multimodal model architectures

  • Experience with inference optimization techniques (KV-cache management, speculative decoding, post-training quantization) and serving frameworks (vLLM, TensorRT-LLM)

We encourage you to apply even if you do not believe you meet every single qualification. If you don't see a role that fits, we encourage you to submit a general application and tell us how you'd like to contribute to our mission.

Similar Jobs

6 Minutes Ago
Hybrid
15-24 Hourly
Junior
15-24 Hourly
Junior
eCommerce • Fashion • Retail • Sales • Wearables • Design
Front-line brand ambassador delivering personalized luxury retail service. Drive sales via omni-channel selling (mobile POS, clienteling, social selling), meet individual and store KPIs, handle transactions, inventory, visual merchandising, and daily store operations. Build client relationships, source customers, support teammates, participate in training and brand initiatives, and maintain store standards and asset protection.
Top Skills: Clienteling ToolsIpadLaptopMobile PosShort-Form VideoSocial Selling PlatformsWalkie-Talkie
33 Minutes Ago
Remote or Hybrid
2 Locations
105K-163K Annually
Senior level
105K-163K Annually
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Manage and grow strategic partnerships with Presidio and Trace3 by developing and executing joint GTM plans, coordinating cross-functional enablement and marketing, leveraging investments to maximize ROI, aligning with sales leadership, and using data-driven insights to drive partner-sourced revenue and brand elevation.
38 Minutes Ago
Remote or Hybrid
USA
123K-228K Annually
Senior level
123K-228K Annually
Senior level
Machine Learning • Payments • Security • Software • Financial Services
Lead and manage engineering teams building scalable, low-latency fraud detection systems. Drive system design, performance optimization, streaming/event-driven data platforms, Agile delivery, regulatory compliance, and talent development while partnering with product and risk stakeholders to improve automation and platform reliability.
Top Skills: Data Management Platform (Dmp)Distributed SystemsEvent-Driven ArchitectureHigh-Throughput SystemsLow-Latency SystemsRule EnginesStreaming

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account