AI Research Engineer
Galileo AI is a pioneering company that operates at the intersection of AI and design with a mission to revolutionize UI design in the age of generative AI. Our state-of-the-art AI enables the creation of interface designs from natural language, empowering people to design beyond their imagination.
Our founding team consists of builders who have previously worked at Google Research, Facebook, and high-growth unicorn startups such as Faire and Cruise. Our team has extensive experience building and designing AI-first products such as GBoard, Google Search, and self-driving technology at Cruise.
We are thrilled to expand our team and are actively seeking talented engineers who are passionate and motivated by our ambitious mission. We strive to create a collaborative environment where individuals can work on groundbreaking technology.
What we're looking for
We're looking for a skilled and motivated AI Research Engineer with deep knowledge in machine learning and NLP to join our team. You'll research and develop novel AI algorithms with large language models. We are looking for someone excited to try new ideas, read the latest papers and drive the model development from research to implementation.
Responsibilities
- Develop new research ideas and algorithms to improve our models
- Effectively manage and track model training experiments to maintain an organized record of results
- Work on our machine learning training codebase to allow for fast iterations and learnings
- Evaluate models and compare results to learn from each experiment
- Process large datasets with parallelization in mind
- Analyze datasets to find insights and track potential bugs
- Stay up-to-date with the latest research in NLP and large language models
- Deploy models to production and integrate with our backend
Experience
- Trained and shipped NLP models to production
- Strong coding skills in Python and knowledge machine learning frameworks (PyTorch, Jax, TensorFlow)
- Worked with large, complex datasets and analyzed data trends and patterns
- Deep knowledge of NLP models (GPT, BERT, T5, etc) and tasks
- Built data pipelines (for example with Airflow, Beam) with a cloud provider (AWS, GCP, Azure)
- Computer vision knowledge (diffusion models, CNN)
- 4+ years of working full-time
- Nice to have: Academic publications in the field of NLP/LLM