Design, implement and evaluate models, agents and software prototypes of large foundational models.
Deep dive into fundamentals of both the ML aspects of foundational models (like architectures, loss functions, data, evals) as well as their implementation on neural accelerators (efficiency during training, serving).
Report and present research findings and developments including status and results clearly and efficiently both internally and externally, verbally and in writing.
Suggest and engage in team collaborations to meet ambitious research goals.
Work with external collaborators and maintain relationships with relevant research labs and key individuals as appropriate.
Work in collaboration with our Responsible AI teams to ensure our advances in intelligence are developed ethically and provide broad benefits to humanity.
Qualifications
PhD in a technical field or equivalent practical experience.
PhD in Machine Learning, Computer Vision, Natural Language Processing or related areas
A proven track record of publications and relevant experience in one or more areas such as large foundational models, ML+Systems, optimization, learning theory, computer vision, NLP etc.
Proven experience with ML frameworks (e.g. JAX) and proven experience with training large models
Proven experience working in industry, working on projects from proof-of-concept through to implementation, applying experimental ideas to applied problems
A real passion for AI, Optimization, and Efficiency!