Research Scientist - Graph Foundation Models

Role Description In this role, you will define and execute an applied research agenda at the frontier of AI. You will build and optimize our entire ML pipeline — from data collection and preparation to training, evaluation, and production monitoring — while developing novel architectures that push the boundaries of what's possible. You'll fine-tune our foundational models for diverse client applications, and work on innovative solutions involving massive datasets and cutting-edge research. Additionally, you'll have the opportunity to work with global AI advisors who are leaders in research worldwide, and join a team that was selected by AWS as one of the three model builders in LATAM. Responsibilities Innovative Model Architecture: Design, implement, and refine new machine learning and deep learning architectures by staying current with academic research and reading relevant papers daily. End-to-End Pipeline Management: Build, test, and monitor robust data pipelines and training processes, ensuring scalability and reliability in production environments. Model Fine-Tuning & Customization: Adapt and optimize our foundational models for various client needs, leveraging experimentation and hyperparameter tuning platforms. Application Development: Develop diverse AI-driven applications that integrate seamlessly with our platform, translating research insights into real-world solutions. Product Integration: Collaborate with engineering and product teams to transform complex business challenges into technical solutions that harness the power of frontier research. Technical Leadership: Take full technical ownership of research projects, making strategic decisions that directly influence the evolution of our products. You Stand Out If You have deep, hands-on experience with PyTorch or similar deep learning frameworks. You are passionate about frontier research, routinely exploring academic literature to drive innovation. You have a track record of working with complex data pipelines and large-scale machine learning systems. You’re experienced in experimenting with and fine-tuning models using advanced hyperparameter tuning platforms. Experience with graph-based models is a plus. You thrive in a dynamic, cross-functional team environment and are comfortable taking full responsibility for your projects. Qualifications Experience: At least 5 years in designing, implementing, and monitoring impactful machine learning models. Academic Background: Bachelor’s degree (preferably with a specialization or Master’s) in Statistics, Computer Science, Mathematics, Physics, Economics, or a related quantitative field. Technical Proficiency: Extensive experience with modern deep learning frameworks (preferably PyTorch) and a solid understanding of state-of-the-art modeling techniques. Research Mindset: Strong ability to design experiments, evaluate model performance, and implement algorithmic improvements based on the latest research. Collaboration: Excellent communication skills and experience working in agile, cross-functional teams. Differentiators Prior experience with distributed systems and processing large volumes of data. Exposure to graph-based models and related technologies. Active involvement in applied AI research, including publications or contributions to the research community. Familiarity with MLOps practices for model deployment and production integration. What We Offer Competitive Compensation: Attractive salary, equity participation, and full transparency in our compensation package. Autonomy and Impact: Work directly with our founders and technical leadership, with the freedom to shape the direction of groundbreaking research. Cutting-Edge Environment: Engage in frontier research, working with huge amounts of data and developing models that have real-world impact across diverse applications. Global Expertise: Benefit from the guidance of global AI advisors and join a team recognized by AWS as one of the three model builders in LATAM. Flexible Culture: Enjoy 100% remote work with access to our São Paulo office, unlimited vacation, and a comprehensive benefits package including a national health plan and extended parental leave.