OnX

Staff Software Engineer

OnX

full-time

Posted on:

Location Type: Remote

Location: Remote • 🇺🇸 United States

Visit company website
✨ AI Apply
Apply

Salary

💰 $210,000 - $215,000 per year

Job Level

Lead

Tech Stack

Distributed SystemsPythonPyTorchRustTensorflow

About the role

  • Design, build, and ship system-level software, frameworks, and tools that unlock performance on advanced hardware.
  • Write Rust-based software that is performant, reliable, and safe, alongside code in C/C++ and Python where needed.
  • Collaborate with hardware teams to co-design solutions that maximize specialized compute architectures.
  • Architect and implement distributed systems, APIs, and libraries that scale across clusters and heterogeneous environments.
  • Dive deep into performance bottlenecks, debugging at the intersection of hardware and software.
  • Mentor engineers, guide design reviews, and help shape technical direction while staying close to the codebase.

Requirements

  • 10+ years of software engineering experience, with a strong track record of shipping production systems.
  • Proficiency in Rust, with deep experience building performance-critical, concurrent, or systems-level applications.
  • Strong background in HPC, GPU/accelerator programming (CUDA, ROCm, OpenCL), or distributed systems.
  • Solid understanding of computer architecture, parallelism, and performance optimization.
  • Familiarity with AI/ML frameworks (e.g., PyTorch, TensorFlow, JAX) and their hardware interactions.
  • Proven track record of building APIs, SDKs, or developer tools used in production environments.
  • Experience mentoring engineers and leading architectural discussions.
  • Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field (PhD a plus).
Benefits
  • 100% company-paid health insurance for employees *and* qualified dependents, because your family’s health matters to us too
  • Flexible Time Off (FTO) to rest, recharge, and take the time you need — no accruals, just trust
  • Company-paid life insurance to provide peace of mind
  • Company contribution to your Flexible Spending Account (FSA) to help with healthcare and dependent care costs
  • Optional add-ons like extra life insurance and critical illness protection if you want additional coverage
  • Paid holidays off so you can truly unplug
  • Work From Anywhere — flexibility to do your best work from wherever life takes you
  • A supportive culture that values balance, wellbeing, and being human

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills
RustCC++PythonHPCGPU programmingCUDAROCmOpenCLAPI development
Soft skills
mentoringcollaborationdesign reviewstechnical directiondebugging
Certifications
Bachelor's degree in Computer ScienceMaster's degree in Computer SciencePhD in Computer Science