A
Senior Machine Learning Performance Engineer
On-siteFull-timeAI & Machine LearningCloud ComputingEnterprise Software
Join a performance engineering team developing software for machine learning accelerators and optimizing workloads across the ML software stack. The role is based in Tel Aviv and focuses on high-performance systems, kernels, SDK capabilities, and end-to-end efficiency.
Responsibilities
- Optimize system performance across the ML software stack
- Analyze high-performance ML workloads on custom accelerators
- Develop high-performance kernels for critical ML operations
- Enhance SDK capabilities and developer experience
- Contribute to instruction scheduling, memory management, parallelism, and compiler improvements
- Design, implement, test, deploy, and maintain software solutions
- Create metrics, implement automation, and resolve software defects
- Participate in design discussions, code reviews, and cross-functional collaboration
Requirements
- Bachelor’s degree in computer science, computer engineering, or a related field
- 6+ years of non-internship professional software development experience
- Proficiency in Python or C++
- Experience with PyTorch, TensorFlow, and/or JAX
- 3+ years of performance engineering experience in accelerated computing
Nice to have
- Master’s degree in computer science or a related field
- Experience with compiler optimization, kernel development, or hardware-software co-design
- Performance optimization experience for LLM, computer vision, or other deep-learning models
Benefits
- Flexible work hours and arrangements
- Inclusive team culture and employee affinity groups
- Mentorship and career-development resources