ML Architect – AI Architecture Research (Equity, Remote)

Jobgether; México; Presencial; MXN 900.000 - 1.500.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Competitive compensation

Equity

Global distributed team

Research-focused environment

Descripción de la vacante

Jobgether is seeking a Machine Learning Engineer - AI Architecture Research based in Mexico. You will explore novel neural architectures beyond Transformers, prototype end-to-end models, and collaborate with inference and systems teams to enable scalable deployment.

You will design experiments, reproduce cutting-edge research, and contribute to both research and production-oriented engineering in a fast-moving environment.

Formación

• Strong foundation in ML and deep learning with practical model development experience.
• Hands-on experience implementing neural network architectures from scratch.
• Strong understanding of attention mechanisms, RNNs, state-space models, or related approaches.
• Knowledge of training dynamics, optimization, scaling, memory, latency, and efficiency.

Responsabilidades

• Research and develop novel neural network architectures, including alternatives to Transformers.
• Design and execute architecture-level experiments focused on scaling laws and trade-offs.
• Prototype models end-to-end for training-ready implementations.
• Analyze model behavior and architectural strengths and limitations.
• Collaborate with inference and systems teams to ensure deployability.
• Read, reproduce, evaluate, and extend cutting-edge ML research.

Conocimientos

Deep learning fundamentals

Neural network architectures

Attention mechanisms

Research and experimentation

Communication skills

Python (ML)

Herramientas

PyTorch

JAX

Descripción del empleo

Jobgether is seeking a Machine Learning Engineer - AI Architecture Research based in Mexico. You will explore novel neural architectures beyond Transformers, prototype end-to-end models, and collaborate with inference and systems teams to enable scalable deployment.

You will design experiments, reproduce cutting-edge research, and contribute to both research and production-oriented engineering in a fast-moving environment.