OpenAI

258 jobs posted

View company profile →

Please mention that you found this job on tryremotely.com. Thanks & good luck!

Tired of Manually Applying to Jobs?

Let JobCopilot do it for you!

Set your preferences and let your AI copilot handle the job search while you sleep.

Applies for jobs that actually match your skills
Tailors your resume and cover letter automatically
Works 24/7—so you don't have to

Activate JobCopilot

Follow us on LinkedIn!

Software Engineer, Trainium

Added

9 days ago

Location

🇺🇸 San Francisco

Type

Full time

Salary

Upgrade to Premium to se...

Related skills

aws gpu llvm compilers kernels

Meet JobCopilot: Your Personal Al Job Hunter

Automatically Apply to Your Dream Jobs While You Sleep

📋 Description

Build and optimize OpenAI's inference stack for AWS Trainium
Develop high-performance kernels for model ops
Extend and improve compiler support for Trainium
Execute and optimize the model forward pass on Trainium
Profile workloads and identify bottlenecks across stack
Collaborate with inference/ML systems teams on new models

🎯 Requirements

3+ years in ML systems, compilers, kernels, runtimes, or perf engineering
Strong systems programming fundamentals
Experience with GPU/TPU/Trainium or similar accelerators
Ability to reason about perf across hardware to ML frameworks
End-to-end ownership of complex performance problems
Bonus: AWS Trainium or Neuron SDK

🎁 Benefits

Equity
Competitive salary
OpenAI benefits package

Apply on employer's website

This employer gathers applications via their own applicant tracking system.

You will be redirected to an external application form.

Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Activate JobCopilot