Remote AI Architecture Research Engineer

Jobgether; México; Remote; MXN 1,200,000 - 2,000,000; Full time

Benefits offered by this job

Equidad competitiva

Trabajo distribuido a nivel mundial

Remoto desde cualquier ubicación

Job summary

Jobgether busca un Ingeniero de Machine Learning orientado a la arquitectura de IA para investigar y diseñar nuevas arquitecturas de modelos, con foco en producción escalable. Colaborarás en proyectos que demandan investigación aplicada y desarrollo práctico en un entorno remoto.

El rol implica explorar alternativas a enfoques Transformer, evaluar trade-offs de memoria, cómputo y latencia, y convertir resultados en prototipos entrenables.

Qualifications

• Fundamentos de ML y DL aplicados a desarrollo de modelos.
• Experiencia práctica implementando arquitecturas de redes neuronales desde cero.
• Comprensión de mecanismos de atención, RNNs y modelos de largo contexto.
• Conocimiento de dinámica de entrenamiento, optimización y escalabilidad de arquitectura.
• Conocimiento sobre memoria, latencia y eficiencia de modelos.
• Dominio de PyTorch o JAX para código de investigación.

Responsibilities

• Investigar y desarrollar arquitecturas neuronales novedosas, incluidas extensiones a Transformers o modelos híbridos.
• Diseñar experimentos a nivel de arquitectura para escalado y rendimiento.
• Prototipar modelos end-to-end y convertir conceptos de investigación en implementaciones entrenables.
• Analizar comportamientos del modelo, fallos y sesgos inductivos.
• Colaborar con equipos de inferencia y sistemas para asegurar eficiencia y despliegue.
• Leer, reproducir y extender papers de ML de vanguardia.
• Contribuir a notas de investigación, benchmarks y benchmarks públicos cuando aplique.
• Alternar entre investigación teórica, experimentación rápida y desarrollo orientado a producción.

Skills

ML basics

Deep learning

Attention mechanisms

RNNs

Hybrid architectures

PyTorch

JAX

Research-oriented coding

Communication skills

Tools

PyTorch

JAX

Job description

Jobgether busca un Ingeniero de Machine Learning orientado a la arquitectura de IA para investigar y diseñar nuevas arquitecturas de modelos, con foco en producción escalable. Colaborarás en proyectos que demandan investigación aplicada y desarrollo práctico en un entorno remoto.

El rol implica explorar alternativas a enfoques Transformer, evaluar trade-offs de memoria, cómputo y latencia, y convertir resultados en prototipos entrenables.