This position is listed on behalf of a partner company, who manages all applications and next steps.

Local

São Paulo - SP

Remoto

Research and develop novel neural network architectures
Design and execute architecture-level experiments
Prototype models end-to-end
Analyze model behavior, failure modes, inductive biases, and architectural strengths and limitations
Collaborate with inference and systems engineering teams
Read, reproduce, evaluate, and extend cutting-edge machine learning research papers
Contribute to internal research notes, benchmarks, experiments, and open-source initiatives where applicable
Move fluidly between theoretical investigation, rapid experimentation, and production-oriented engineering

Strong foundation in machine learning and deep learning fundamentals
Hands-on experience implementing neural network or model architectures from scratch
Strong understanding of attention mechanisms, RNNs, state-space models, hybrid architectures, or related approaches
Solid knowledge of training dynamics, optimization, scaling behavior, and architecture-level performance considerations
Understanding of model-level memory, latency, compute, and efficiency constraints
Proficiency with PyTorch or JAX and the ability to develop and experiment with research-oriented ML code
Ability to evaluate architectural ideas through both theoretical reasoning and empirical experimentation
Strong communication skills, with the ability to clearly explain technical concepts and architectural trade-offs

Competitive compensation and meaningful equity
Opportunity to work directly on core AI model architecture
Significant influence over technical and research direction within a rapidly growing organization
Small, high-caliber team with fast feedback loops and a strong research-oriented environment
Opportunity to take research concepts from experimentation through to production deployment
Full-time position with a globally distributed work environment

This position is listed on behalf of a partner company, who manages all applications and next steps.

Machine Learning Engineer — AI Architecture Research