Staff Engineer, ML Profiling Tools

Home > Python programming jobs

Staff Engineer, ML Profiling Tools in USA new

Samsung Semiconductor
Full time
Email

USA
10/04/2026
Report

San Jose, CA

Responsibilities

Develop and maintain large-scale, interactive web-based visual analytics tools for AI/ML workloads.
Build fluid data visualizations such as flame graphs, call trees, and timeline charts for massive multidimensional telemetry datasets.
Partner with compiler engineers and hardware architects to translate performance-profiling metrics into clear developer workflows.
Optimize the developer experience for machine learning performance profiling and help engineers diagnose inference bottlenecks.
Communicate with users, partners, and management to deliver systems on time and within budget.

Requirements

Bachelor’s degree plus 10+ years, master’s degree plus 8+ years, or PhD plus 5+ years of industry experience.
Extensive experience designing and implementing rich interactive data visualizations and user interfaces with modern web technologies.
Experience or strong interest in developer tools, IDE extensions, or complex diagnostic dashboards.
Ability to turn massive, unstructured, or multidimensional telemetry and profiling data into intuitive visual representations.
Experience designing workflows for software engineers, data scientists, or researchers that reduce cognitive load and streamline root-cause diagnosis.
Hands-on experience with or strong curiosity about systems profiling and tracing tools such as Perfetto, Chrome Tracing, NVIDIA Nsight Systems, NVIDIA Nsight Compute, PyTorch Profiler, TensorBoard, and eBPF.
Understanding of CPUs, GPUs, TPUs, interconnects, networking, and common performance bottlenecks.
Working knowledge of PyTorch, JAX, TensorFlow, LLM inference, pipeline parallelism, tensor parallelism, or kernel dispatch is a bonus.
Strong problem-solving, critical-thinking, collaboration, communication, curiosity, and adaptability skills.

Benefits

Daily onsite work at the San Jose, California office under the company’s flexible work policy.
Base pay range of $163,000–$253,000 USD plus incentive opportunities tied to individual and company performance.
Medical, dental, vision, 401(k), charitable giving match, and 4+ weeks of paid time off annually plus holidays and sick leave.
Family-support benefits including fertility or adoption stipends, medical travel support, and virtual veterinary care.
Emotional wellness support including on-demand apps and free confidential therapy sessions.
Onsite café and gym, virtual fitness classes, and a flexible work environment.

Salary: $163k - $253k/yr

Samsung Semiconductor

Samsung Semiconductor is a leading innovator in advanced semiconductor solutions, delivering high-performance chips that power everything from consumer electronics to next‑generation data centers. By combining cutting‑edge research with scalable manufacturing, the company enables faster, more eff...

Similar positions

Staff Data Engineer

NABIS
Full time

USA
10/04/2026
Salary: $145k-$170k

Remote

Principal Software Engineer, Security & Compliance

Thomson Reuters
Full time

Others
10/04/2026

Zug, Switzerland

Senior AI ML Engineer

Viatris
Full time

Others
10/04/2026

Bengaluru, India

Software Quality Engineer, Radar

Anduril Industries
Full time

USA
10/04/2026
Salary: $111k - $147k/yr

Fort Collins, CO

Staff AI Engineer - Architecture skills

Shift Technology
Full time

Others
10/04/2026
Salary: Competitive

Paris, France