onsite
Research Scientist, Gemini Data
Research Engineer
Research Scientist focused on advancing Gemini data pipelines and large language models, applying JAX and Python to design, train, and fine‑tune state‑of‑the‑art neural networks for high‑impact machine‑learning research.
About the role
Key Responsibilities
- Design and implement scalable data processing pipelines for training large language models using JAX.
- Develop novel fine‑tuning techniques to improve model performance on domain‑specific tasks.
- Conduct rigorous experiments, analyze results, and publish findings in top‑tier venues.
- Collaborate with cross‑functional research teams to integrate model improvements into broader AI systems.
- Mentor junior researchers and contribute to open‑source tooling for reproducible research.
Requirements
- Ph.D. or equivalent experience in Machine Learning, Computer Science, or a related field.
- Strong proficiency in Python and JAX for high‑performance model development.
- Hands‑on experience with large language models, including training, evaluation, and fine‑tuning.
- Demonstrated ability to design and execute complex research experiments.
- Excellent communication skills and a track record of publishing research results.
Skills
pythonmachine learning