Staff AI Application Engineer - Heterogeneous Computing

Esta es tu oportunidad de dar forma y formar parte de una organización internacional, dinámica y de rápido crecimiento. Ofrecemos un paquete de compensación atractivo, que incluye un plan de pensiones, seguros de empleados extensos y la opción de obtener acciones de la empresa.

Te espera una cultura abierta que apoya la creatividad y la innovación continua. La propiedad colaborativa y la libertad con responsabilidad son características de nuestra forma de actuar y trabajar en equipo.

En Axelera AI, adoptamos de todo corazón la igualdad de oportunidades y valoramos la diversidad de la manera más alta. Nuestro compromiso inquebrantable es cultivar un entorno cálido e inclusivo que empodere y celebre a cada miembro de nuestro equipo. Damos la bienvenida a solicitantes de todos los orígenes para que se unan a nosotros en la configuración del futuro de la IA.

Como Staff AI Application Engineer en el equipo de Model Enablement (Application Software), construirás nuestra plataforma de habilitación de modelos y decidirás cómo se ejecuta cada modelo a través de la Metis o Europa AIPU, la CPU host y los aceleradores del host. La forma en que este trabajo se divide y se programa determina la mayor parte del rendimiento de extremo a extremo que un cliente experimenta en el Voyager SDK.

El rol combina arquitectura y trabajo práctico en el frontend del modelo, el compilador de grafos, el runtime y el pipeline de la aplicación. Durante la fase de implementación del modelo (model bring-up), te encargarás del frontend y la parte del grafo, mientras que un ingeniero de la parte del dispositivo cubrirá el backend.

Nuestro equipo utiliza agentes de codificación de IA en el trabajo diario, y tú los utilizarás para construir, probar y verificar tu propio trabajo. También damos la bienvenida a ingenieros de aplicaciones de campo (field application engineers), líderes técnicos y responsables de ingeniería experimentados que quieran volver a la ingeniería práctica. La experiencia en planificación, delegación y revisión de tareas se transfiere muy bien al trabajo con agentes de IA.

• Plataforma de habilitación de modelos y entorno de agentes: Construir la plataforma que los ingenieros y los agentes de IA utilizan para implementar modelos: herramientas reutilizables para particionar, compilar, ejecutar y perfilar; entornos con acceso a hardware real; ejecuciones de bucle cerrado que aplican un cambio, lo miden frente a la referencia y repiten; y verificación (pruebas diferenciales, detección de regresiones) integrada en CI.

• Ingesta de modelos: Responsabilizarte de cómo se captura, analiza y divide un modelo PyTorch o ONNX entre las unidades de matriz y vector RISC-V de la AIPU, la CPU host y los aceleradores del host (GPU, NPU, decodificador de vídeo), con etiquetas de ubicación que los usuarios puedan leer y anular. Construir esto junto al equipo del compilador de grafos para que funcione con cada modelo.

• Ejecución heterogénea: Extender esa división a modelos grandes, cascadas y pipelines multi-modelo, por ejemplo, mediante la descarga de expertos (expert offloading) y el almacenamiento en caché entre la memoria del dispositivo y la del host. Acordar el plan con el equipo del compilador de grafos.

• Rendimiento de extremo a extremo: Junto con el equipo de aplicaciones (constructor de pipelines) y el equipo de runtime, lograr que cada modelo alcance su mejor rendimiento de extremo a extremo: solapamiento del trabajo del host y del dispositivo, buffers zero-copy, batching, programación multi-núcleo y modelos con múltiples entradas. Perfilar el tiempo del dispositivo, el tiempo del host y el movimiento de datos, y confirmar cada aceleración en el pipeline completo.

• Modelos críticos: Estudiar los modelos que más importan a los clientes, tanto en Metis hoy como en Europa, y convertir cada lección en una mejora de la plataforma.

Ofrecemos un acuerdo de trabajo flexible, con opciones para:

• Trabajar desde una de nuestras oficinas de Axelera AI (Leuven en Bélgica, Ámsterdam y Eindhoven en los Países Bajos, Zúrich en Suiza, Florencia y Milán en Italia o Bristol en el Reino Unido) si ya resides en las cercanías.

• Trabajar totalmente de forma remota desde cualquier país europeo (incluido el Reino Unido) en el que ya te encuentres.

• Relocalizarte con nosotros y trabajar desde Italia (Florencia o Milán) o los Países Bajos (Ámsterdam o Eindhoven).

Ten en cuenta que se dará prioridad a los candidatos que tengan [interés en estar] basados en Bélgica o Italia.