Staff Engineer, AI/ML Visualization

Position Overview

Advancing the World's Technology Together – Our AGI Computing Lab is dedicated to solving the complex system-level challenges of future AI/ML workloads. We design and develop scalable platforms that effectively handle the enormous computational and memory demands while minimizing energy consumption and maximizing performance. By collaborating with hardware and software engineers, our team is continuously researching and developing emerging technology trends in memory, computing, interconnect, and AI/ML.

We believe that innovation and growth are driven by an inclusive culture, and together we are building a better tomorrow for our employees, customers, partners, and communities.

Location: Daily onsite presence at our San Jose, CA office / U.S. headquarters in alignment with our Flexible Work policy.

Salary: $163,000—$253,000 USD

Key Responsibilities

• Develop and maintain highly interactive, large-scale, and complex web-based visual analytics tools on AI/ML workloads.
• Design and implement creative and performant visualizations to assist in diagnosing issues or uncovering optimization opportunities in AI/ML inference and serving.
• Collaborate closely with hardware architects and compiler engineers to define, design, and implement effective visualization solutions.
• Develop high-performance visualization workflows by integrating independent graphics modules with micro-services for asset optimization, ensuring a non-blocking client experience.
• Continuously seek improvements to existing tools and devise new approaches to meet evolving requirements.
• Stay up-to-date with the latest advancements in web development and visualization techniques.
• Contribute to the development of documentation and user guides.
• Perform other assigned responsibilities.

Required Qualifications

• Bachelor's with 10+ years, or Master's with 8+ years, or PhD with 5+ years of industry experience.
• Proven experience in full-stack visual analytics software development, specifically building scientific or infrastructure visualization tools; experience in AI model visualization integrated with PyTorch is a plus.
• Strong proficiency in modern frontend technologies like React and backend technologies such as Node.js or Python-based frameworks; experience with graph-based UI is a plus.
• Experience with WebGL/WebGPU frameworks such as three.js or deck.gl; knowledge of GLSL or WebAssembly with C++ is a plus.
• Excellent problem-solving skills with the ability to think critically and creatively.
• Strong communication and interpersonal skills, with an ability to work independently and collaboratively within a team.
• A drive to clarify deep technical complexities while balancing creative visualization with engineering rigor.
• Demonstrated ability to adapt to diverse styles and global norms, fostering an inclusive work environment.
• A passion for learning, collaboration, and innovating new approaches in a fast-paced environment.

Benefits & Perks

• Give Back – Enjoy a charitable giving match and frequent opportunities to get involved, supporting the community.
• Enjoy Time Away – Start with 4+ weeks of paid time off annually, plus holidays and sick leave to rest and recharge.
• Care for Family – Receive support including a stipend for fertility care or adoption, medical travel support, and virtual vet care.
• Prioritize Emotional Wellness – Access on-demand apps and free confidential therapy sessions for your wellbeing.
• Stay Fit – Benefit from an onsite café, gym, and virtual fitness classes that promote a healthy lifestyle.
• Embrace Flexibility – Work within a flexible environment that helps you find the right balance between professional and personal life.