Member of Technical Staff, Kernels
MirendilAI Research company
San Francisco, United States$300K - $500KLead
Andreessen Horowitz
Kleiner Perkins
Nvidia
Data & AI
About the role
TL;DR
Design, implement, and optimize custom ML kernels for model development.
- •We are looking for an engineer to design, implement, and optimize custom ML kernels that bolster our model development stack.
- •Key Responsibilities Design and implement custom performant ML kernels that work at scale Identify inefficiencies and optimize compute-intensive workloads to reduce memory bandwidth bottlenecks and improve hardware utilization Enable and validate low-precision arithmetic formats and contribute to related compiler or runtime stacks Requirements Deep understanding of ML kernels and hardware optimization Experience with TensorFlow, PyTorch, and scikit-learn Strong background in computer vision and NLP Ability to work at the intersection of hardware and frontier AI research
Required skills
TensorFlowPyTorchscikit-learnComputer VisionNLPMLflowKubeflowSageMakerVertex AIONNXJAXKubernetesCI/CDDockerAWS
Domain expertise
ai
Benefits & perks
equity, competitive benefits
Tech stack
TensorFlowPyTorchscikit-learnHugging FaceComputer VisionNLPLLMsMLflowKubeflowSageMakerVertex AIONNXJAXKubernetesCI/CDDockerAWSGoogle CloudAzureTerraformCloudFormationPulumiEC2S3RDSECSEKSGKECloud RunBigQuery