Senior AI Platform Engineer: LLM/VLM Enablement & SDK

Axelera AI; United States; À distance; USD 140 000 - 210 000; Plein temps

Avantages offerts par ce poste

Pension

Insurance

Stock options

Résumé du poste

Axelera AI is a deep-tech company advancing the next-generation AI platform from research to production in the United States. The role focuses on studying new LLM/VLM architectures and prototyping them on hardware, turning results into supported SDK features with stable APIs, tests, and documentation.

Daily work centers on Python and PyTorch, with integration of LLM/VLM serving into the pipeline builder, including KV-cache management and multimodal inputs.

Qualifications

• 5+ years of software engineering experience with ML focus.
• Strong Python and PyTorch skills required.
• Experience with AI coding agents and ML tooling.

Responsabilités

• Study new LLM and VLM architectures and prototype them on hardware.
• Productize results into SDK features with stable APIs, tests and docs.
• Integrate LLM/VLM serving into the pipeline builder with KV-cache management and multimodal inputs.

Connaissances

Python

PyTorch

AI coding agents

Software engineering

Transformer inference

Description du poste

Axelera AI is a deep-tech company advancing the next-generation AI platform from research to production in the United States. The role focuses on studying new LLM/VLM architectures and prototyping them on hardware, turning results into supported SDK features with stable APIs, tests, and documentation.

Daily work centers on Python and PyTorch, with integration of LLM/VLM serving into the pipeline builder, including KV-cache management and multimodal inputs.