AI Research Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Code Metal · 2 months ago

AI Research Engineer

Code Metal is focused on enabling hardware deployment at the speed of software development. In this role, you will collaborate with a small team of talented researchers on ambitious projects in generative AI and reinforcement learning.

Artificial Intelligence (AI)Information TechnologySoftware

Responsibilities

Design, execute, and analyze experiments with a high degree of independence
Contribute to core models and frameworks
Create high-quality datasets (both in-the-wild and synthetic)
Perform literature reviews and implement new techniques from papers
Contribute to publications, present at conferences and workshops, etc

Qualification

LLM research experiencePython programmingPyTorchHuggingFacePublications in conferencesLarge-scale LLM trainingPh.D. in Computer ScienceDistributed computing experience

Required

Ph.D. in Computer Science or a closely related field
Prior LLM research experience
Comfortable programming in Python and familiar with frameworks such as PyTorch and HuggingFace

Preferred

Publications at peer-reviewed conferences such as NeurIPS, ICLR, ICML, etc
Experience with large-scale LLM training, particularly in a distributed computing environment

Benefits

Competitive salary
Health care plan (medical, dental, and vision)
Retirement plan (401k, IRA) with employer matching
Unlimited PTO
Flexible hybrid work arrangement
Relocation assistance for qualifying employees

Company

Code Metal

twittertwitter
company-logo
Code Metal delivers an AI-based development platform that transforms high-level code into edge-optimized, production-ready deployments.

Funding

Current Stage
Growth Stage
Total Funding
$59.45M
Key Investors
AccelShield CapitalJ2 Ventures
2025-11-12Series A· $36.5M
2025-06-26Series Unknown· $6.5M
2024-07-23Seed· $13M
Company data provided by crunchbase