Java engineer Jobs in Berkeley, CA

Software Engineer – High Performance Systems / ML Infrastructure

Location: Mountain View, CA (Hybrid – 3 days onsite)

Make your application after reading the following skill and qualification requirements for this position.

A well-funded, high-growth deep tech company is building next-generation compute infrastructure to support large-scale machine learning workloads.

This is a rare opportunity to work across the full stack — from hardware interaction through to system-level performance — in a team focused on solving some of the hardest problems in compute.

What you’ll be doing

• Build and maintain functional and performance models of custom hardware
• Develop tools to support debugging, validation, and performance analysis
• Work closely with architecture, compiler, runtime, and validation teams
• Contribute to system and architectural decisions
• Identify and implement performance improvements across the stack

What they’re looking for

• Strong software engineering fundamentals with a focus on performance and maintainability
• Solid understanding of computer architecture
• Experience working in low-level or systems environments
• Ability to work cross-functionally with hardware and software teams

Nice to have:

• Rust experience
• Experience with simulators or hardware modeling
• Exposure to ML infrastructure or accelerator architectures
• Familiarity with performance optimization and benchmarking

Why xywuqvp this role stands out

• You’ll be working on real, hard engineering problems at the intersection of hardware and software
• The team is small, highly technical, and moves quickly
• High level of ownership and impact on core system design
• Strong compensation with flexibility between cash and equity

Compensation

• Broad range depending on experience:
• $220k – $400k base + equity
• Flexible cash vs equity structure