[Remote] Member of Engineering (Pre-training / Data Research)

Note: The job is a remote job and is open to candidates in USA. Poolside is a pioneering company focused on creating Artificial General Intelligence. The role involves working on the data team to enhance the quality of datasets for training models through data generation and optimization, collaborating with various teams to meet data needs.ResponsibilitiesFollow the latest research related to LLMs and data quality in particular. Be familiar with the most relevant open-source datasets and modelsDesign and implement complex pipelines that can generate large amounts of data while maintaining high diversity and optimizing the resources availableClosely work with other teams such as Pretraining, Posttraining, Evals and Product to ensure short feedback loops on the quality of the models deliveredSuggest, conduct and analyze data ablations or training experiments that aim to improve the quality of the datasets generated via quantitative insightsSkillsStrong machine learning and engineering backgroundExperience with Large Language Models (LLM), including: Understanding of transformer architectures and how LLMs learn, Data ablations and scaling laws, Mid-training and Post-training techniques, Training reasoning and agentic modelsExperience with evals tracking model capabilities (general knowledge, reasoning, math, coding, long-context, etc)Experience in building trillion-scale pretraining datasets, and familiarity with concepts like data curation, deduplication, data mixing, tokenization, curriculum, impact of data repetition, etcExcellent programming skills in PythonStrong prompt engineering skillsExperience working with large-scale GPU clusters and distributed data pipelinesStrong obsession with data qualityResearch experience: Author of scientific papers on any of the topics: applied deep learning, LLMs, source code generation, etc. - is a nice to haveCan freely discuss the latest papers and descend to fine detailsIs reasonably opinionatedBenefitsFully remote work & flexible hours37 days/year of vacation & holidaysHealth insurance allowance for you & dependentsCompany-provided equipmentWell-being, always-be-learning & home office allowancesFrequent team get togethersDiverse & inclusive people-first cultureCompany OverviewPoolside is an artificial intelligence platform that offers foundation concepts and infrastructure to write software codes. It was founded in 2023, and is headquartered in San Francisco, California, USA, with a workforce of 51-200 employees. Its website is http://www.poolside.ai.Company H1B SponsorshipPoolside has a track record of offering H1B sponsorships, with 1 in 2025. Please note that this does not guarantee sponsorship for this specific role.