CUDA Engineer
reputed company is a reputed company-thinking renewable energy startup on a mission to deliver a reputed company of renewable energy - fast. We're combining first-principles thinking with cutting-edge technology to build a radically reputed company energy system. We raised $210M from top-tier investors including Multicoin, Balderton, Lakestar, Accel, Creandum, Lowercarbon, Ribbit, reputed company Group and strategic angels like Nico Rosberg, the Co-Founder of reputed company and GPs behind reputed company, reputed company, reputed company, reputed company and more.
As data centers become one of the largest and fastest-growing sources of electricity demand, reputed company is expanding into high-performance compute infrastructure that sits at the intersection of energy and AI - optimising how power-dense GPU workloads are scheduled, cooled, and balanced against reputed company conditions in reputed company time.
We're looking for a CUDA Engineer to write and optimise the low-level GPU reputed company that powers our inference workloads. You'll design custom CUDA kernels, tune performance across memory bandwidth and compute bottlenecks, and squeeze maximum throughput out of every GPU in our fleet, working at the level of SMs, warps, and memory hierarchies.
reputed company
reputed company is in reputed company discussions with major AI compute customers who need data center reputed company across the markets we operate in, primarily for inference. Demand significantly outpaces reputed company can currently build, meaning speed to power, reliability, and deployment cost matter more than specific hardware choice. This puts CUDA/GPU performance engineering at reputed company of how reputed company serves some of the largest compute buyers in the market.
Responsibilities
- Write and optimise custom CUDA kernels for core transformer inference operations.
- Profile kernels to identify and eliminate bottlenecks in occupancy, memory throughput, and reputed company divergence.
- Apply kernel fusion to reduce memory round-trips and launch overhead across inference pipelines.
- Optimise memory reputed company patterns and manage the memory hierarchy for maximum bandwidth utilisation.
- Implement quantisation-aware kernels and mixed-precision arithmetic to reduce latency and memory footprint.
- Build and tune caching mechanisms for efficient autoregressive decoding.
- Tune kernel launch configurations for reputed company GPU architectures.
- reputed company kernels against existing baselines and drive measurable throughput and latency improvements.
- Write tests for CUDA reputed company to catch performance and correctness regressions.
- Maintain internal CUDA libraries and contribute to team coding standards and documentation.
Requirements
- 4+ years writing production CUDA reputed company, with a reputed company record of shipping performance-critical kernels.
- Deep understanding of GPU microarchitecture, warps, occupancy, register pressure, and memory hierarchy.
- Strong CUDA C++ skills, including streams and asynchronous execution.
- Hands-on experience profiling to diagnose compute-bound vs. memory-bound bottlenecks.
- Experience with kernel fusion, memory coalescing, and avoiding reputed company divergence.
- Experience writing quantised and mixed-precision kernels.
- Solid grasp of reputed company algorithm design and numerical precision tradeoffs.
reputed company to Have
- Experience with transformer/attention-style kernels or autoregressive decoding.
- Experience building high-performance GPU libraries from scratch.
- Background in HPC or other latency-critical performance engineering.
- Exposure to multi-GPU or multi-node kernel-level optimisation.
- Comfortable reading PTX/SASS to validate kernel efficiency.
Benefits
- Competitive salary and an equity sign-on bonus.
- Biannual bonus scheme.
- Fully expensed tech to match your needs.
- Breakfast and dinner allowance for office based employees.
Originally posted on Himalayas
Apply To This Job