A
Machine Learning Performance Engineer
·Israel
On-siteFull-timeAI & Machine LearningCloud ComputingArtificial Intelligence
Join a team building the Neuron SDK for custom ML accelerators. Drive performance optimization across the entire ML stack, develop high-performance kernels, and enhance compiler and SDK capabilities for LLM, Vision, and other deep learning models.
Responsibilities
- Optimize system performance across the entire ML software stack
- Analyze high-performance ML workloads running on custom hardware
- Develop high-performance kernels for critical ML operations
- Enhance the Neuron SDK to improve developer experience and system capabilities
- Collaborate across Compiler, Frameworks, and Hardware teams to maximize end-to-end performance
- Build high-impact solutions for a large customer base
- Participate in design discussions, code review, and communicate with stakeholders
Requirements
- B.S. or M.S. in computer science or related field
- Proficiency with Python or C++
- Experience with TensorFlow, PyTorch, or JAX
- 3+ years of professional software development experience
- 3+ years of experience in performance optimizations in deep learning models
Nice to have
- M.S. in computer science or related field
- Experience developing algorithms for simulation tools
- Experience with compiler optimization, kernel writing, or hardware-software co-design
Benefits
- Flexible work hours and arrangements
- Mentorship and career growth opportunities
- Inclusive team culture with affinity groups