Member of Technical Staff, Data Infrastructure

• Design, build, and operate scalable, fault-tolerant infrastructure for LLM research: distributed compute, data orchestration, and storage across modalities.
• Develop high-throughput systems for data ingestion, processing, and transformation — including training data catalogs, deduplication, quality checks, and search.
• Build systems for web crawling, data ingestion, and real-time data processing to support model training operations.
• Develop tools and frameworks for efficient data storage, retrieval, and versioning across distributed systems.
• Ensure data collection adheres to privacy regulations.

• BS/MS/PhD in Computer Science, Machine Learning, or a related field (or equivalent experience).
• 3+ years of experience building data processing pipelines at scale, particularly with AI/ML applications.
• Strong proficiency in Python and experience with data processing frameworks (Apache Spark, Beam, Airflow).
• Familiarity with synthetic data generation techniques and data augmentation strategies.
• Familiarity with web scraping, crawling technologies, and Common Crawl datasets.
• Solid understanding of machine learning fundamentals and experience with ML frameworks (PyTorch, TensorFlow).
• Experience with SQL and NoSQL databases for managing structured and unstructured data.

• Experience with large language models and understanding of tokenization, embeddings, and model architectures.
• Experience managing human annotation workflows and quality control processes.
• Experience with vector databases and embedding-based retrieval systems.
• Knowledge of data privacy regulations and ethical AI practices.
• Experience with distributed computing and large-scale data storage systems (HDFS, S3, BigQuery).

Compensation

• Work with World-Class Talent: Collaborate with the inventors of diffusion models and leading AI researchers
• Shape Foundational Technology: Your decisions will influence how the next generation of AI products are built and used
• Immediate Impact: Join at the ground floor where your contributions directly shape product direction and company trajectory

• Competitive salary and equity in a rapidly growing startup
• Flexible vacation and paid time off (PTO)
• Health, dental, and vision insurance
• 401k match
• Catered meals (breakfast, lunch, & dinner)
• Commuter subsidies
• A collaborative and inclusive culture