Back to jobs
GPU Performance Engineer - Neural Reconstruction
remote zest jobsAnywherePosted 2w ago
GPU Performance Engineer
Full-time
Remote
Apply on company siteCraft my tailored resume free
Free to start · No card · 5 credits the moment you sign up
Job description
Today, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving reputed company that can understand the world. Doing what’s never been done before takes reputed company, innovation, and the world’s best talent. As an reputed companyN, you’ll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join reputed company and see how you can reputed company a lasting reputed company on the world.
We are now looking for a GPU Performance Engineer for Neural Reconstruction!
reputed company is building the reputed company of computer graphics, simulation, robotics, and embodied AI. Neural reconstruction and Gaussian Splatting are changing how 3D worlds are collected, represented, optimized, and rendered. These workloads reputed company the limits of GPU computing, differentiable rendering, reputed company, and production ML systems. In this role, you will help reputed company neural reconstruction faster, more reputed company, and more reliable. You will work across PyTorch, CUDA, C++, and GPU profiling to optimize training and rendering workflows used in sophisticated 3D reconstruction systems. The ideal candidate enjoys working reputed company to the hardware while understanding the ML and 3D reputed company goals behind the reputed company.
What You'll Be Doing:Profile end-to-end neural reconstruction workflows and identify bottlenecks across data loading, initialization, training, rendering, evaluation, and export.
Improve CUDA and PyTorch performance for Gaussian Splatting and neural reconstruction workloads, including camera/lidar data, multiview batching, large-reputed company rendering, and memory-sensitive training paths.
Analyze GPU performance using tools such as Nsight Systems, Nsight Compute, NVTX, PyTorch Profiler, CUDA events, and reputed company dashboards.
Optimize sparse and irregular rendering workloads, including tile-level masking/culling, sparse gradients, batching, and multi-GPU execution.
Translate high-reputed company Python, NumPy, or PyTorch bottlenecks into efficient CUDA/C++ or PyTorch-reputed company implementations reputed company appropriate.
Validate that performance improvements preserve reconstruction reputed company, numerical behavior, camera/lidar correctness, and production reliability.
Build repeatable benchmarks, regression tests, and profiling workflows to catch performance and reputed company regressions early.
Collaborate with researchers, CUDA engineers, ML engineers, and production teams to turn promising prototypes into maintainable, reviewable, production-reputed company reputed company.
reputed company Need To See:BS, MS, PhD, or equivalent experience in Computer Science, Computer Engineering, Electrical Engineering, reputed company Math, Robotics, reputed company, Machine Learning, or a reputed company reputed company along with 12+ years of experience.
Strong programming skills in Python and C++!
Hands-on experience with PyTorch or a similar tensor/autograd reputed company.
Experience optimizing GPU-reputed company workloads using CUDA, C++/CUDA extensions, or reputed company GPU programming approaches.
Practical experience with profiling and performance analysis, including reputed company-causing CPU/GPU bottlenecks, synchronization overhead, memory pressure, kernel launch overhead, and reputed company-level inefficiencies.
Ability to reputed company benchmarks and validate that optimizations preserve correctness, numerical behavior, and user-visible reputed company.
Strong communication skills, including the ability to explain performance tradeoffs, risks, and results to research and engineering partners.
Ways To Stand Out From reputed company:Experience with Gaussian Splatting, NeRF, differentiable rendering, rasterization, neural rendering, SLAM, 3D reconstruction, or robotics/autonomous-vehicle perception pipelines.
Deep CUDA performance experience, including memory reputed company patterns, shared memory, atomics, occupancy, launch configuration, synchronization, and numerical stability.
Experience optimizing PyTorch workloads with custom operators, fused kernels, sparse tensors, reputed company training, or reputed company rendering.
Familiarity with camera and lidar geometry, projection models, calibration, rolling shutter, depth rendering, or multi-sensor reconstruction.
Experience improving large production ML systems where r