Backed by Lux.
About the role
Backed by top-tier investors and led by industry veterans, we’re scaling rapidly. We seek forward-thinking engineers and operators who thrive in a collaborative environment and excel at solving complex challenges from first principles. If you want to build infrastructure that fundamentally changes what the world can accomplish with AI, come join us! Found on 1752vc Careers, the job board for startup and VC roles.
What they're looking for
- Design and implement high-performance compute kernels for AI primitives such as GEMM, attention, normalization, and convolution
- Optimize for throughput, latency, and memory hierarchy across heterogeneous compute units (SIMD, matrix engines, DMA)
- Collaborate with compiler and runtime teams to integrate kernels into Triton and PyTorch pipelines
- Profile and tune kernels using tools like Perfetto, VTune, Tracy, Nsight, or custom simulators
- Prototype and evaluate precision formats (FP16/BF16/FP8/e5m2, MXFP/FP4, etc.)
- Contribute to micro-architecture feedback loops, helping co-design ISA and memory features with the hardware team
More about this role
We’re a fast-moving, US-Israeli AI startup building next-generation infrastructure for the world’s most demanding AI workloads. Our mission is to accelerate the future of intelligence and make it ubiquitously accessible by delivering full stack custom AI servers optimized for large-scale inference and training.
Backed by top-tier investors and led by industry veterans, we’re scaling rapidly. We seek forward-thinking engineers and operators who thrive in a collaborative environment and excel at solving complex challenges from first principles. If you want to build infrastructure that fundamentally changes what the world can accomplish with AI, come join us!
About the position
- Design and implement high-performance compute kernels for AI primitives such as GEMM, attention, normalization, and convolution.
- Optimize for throughput, latency, and memory hierarchy across heterogeneous compute units (SIMD, matrix engines, DMA).
- Collaborate with compiler and runtime teams to integrate kernels into Triton and PyTorch pipelines.
- Profile and tune kernels using tools like Perfetto, VTune, Tracy, Nsight, or custom simulators.
- Prototype and evaluate precision formats...
Browse similar: AI jobs · AI startup jobs · Startup jobs