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.

Create profile and match

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