Annapurna Labs Ltd.
Manual apply20 days ago
Senior ML Software Engineer, Data Plane
Onsite · IL, Tel Aviv
pytorch
software development
amazon
Match & cover letter
Create a profile to see your match and get a cover letter for this role.
Source: amazon
Never pay or share passwords or codes with an employer. How to spot a fake job
AI TL;DR
You will design and implement the inference data plane for large models at Annapurna Labs in Tel Aviv. This role involves optimizing low-level code for custom hardware and validating model architectures end-to-end.
AI match
Clear role · 80/100 against the market band on this page.
Interview prep for this job
- What would you ship in the first month as Senior ML Software Engineer, Data Plane at Annapurna Labs Ltd.?
- Which tradeoff in the description would you ask about first?
- How do you explain a production incident to this team?
- What would you measure in the first two weeks?
- Which part of your last project maps to this stack?
Five questions are free. The rest are Pro. See Pro
Similar roles
- Senior Forward Deployed Engineer (Model Customization and Optimization) Amazon Web Services, Inc.
- Research Scientist Pindrop
- Machine Learning Engineer Hestus, Inc.
- Senior Machine Learning Engineer Runware
- Senior AI Engineer Yuno
- Applied Research Scientist, AI Research Descript
Job description
You will design and implement the inference data plane for large models at Annapurna Labs in Tel Aviv. This role involves optimizing low-level code for custom hardware and validating model architectures end-to-end.
Details
- Tel Aviv, work from office
- Skills: PyTorch
- Experience: Senior level
The work
- Develop and optimize compute kernels for ML accelerator architecture
- Implement and validate LLM architectures end-to-end
- Integrate custom accelerator backends into ML serving frameworks
- Build and maintain test infrastructure for model validation