Principal Staff Engineer, AI Platform Research, Data Science

This is a fully remote position, open to applicants in United States.

📋 Description

• Design, implement, and enhance exabyte-scale data platforms and pipelines that support LLMs, neural networks, RAG, and AI-driven systems.

• Foster the adoption and execution of agentic workflows and harnessing techniques for autonomous, data-driven security features.

• Develop production-ready, scalable, fault-tolerant, cost-effective, and thoroughly tested code.

• Provide technical guidance in data modeling, normalization, and semantic cataloging for AI/ML workloads.

• Establish best practices for MLOps/DataOps for LLMs, including monitoring, observability, and zero-touch recovery.

• Mentor engineers through technical workshops and design evaluations.

• Collaborate with Data Scientists, Product Managers, and engineering teams to convert research prototypes into production-level services.

• Oversee development, testing, deployment, and monitoring throughout the entire lifecycle of critical data services.

⛳️ Requirements

• Bachelor’s, Master’s, or PhD in Computer Science, Data Engineering, or a related STEM discipline, or equivalent hands-on experience.

• Over 5 years of progressive experience in Data Engineering or Platform Engineering.

• A minimum of 3 years dedicated to architecting and constructing AI/ML or Data Science platforms at a large scale.

• Practical experience in LLM engineering, including fine-tuning, prompt engineering, and deployment.

• Familiarity with RAG and the development of agentic workflows.

• Demonstrated expertise in designing and delivering large-scale distributed systems, including sharding, partitioning, and concurrency management.

• Expert-level knowledge in at least one of the following: Python, Go, Rust, or JVM technologies.

• Experience with MLflow, SageMaker, Vertex AI, LangChain, or LlamaIndex.

• Proficiency in Spark, Dask, or Flink.

• Knowledge of AWS, GCP, or OCI and associated data services.

• Experience with Docker and Kubernetes.

• Familiarity with Kafka or Pulsar.

• Experience with Snowflake, BigQuery, Airflow, or Kubeflow.

• Ability to produce clean, elegant, efficient, and well-tested code.

• Capability to utilize AI-assisted secure code development.

• Understanding of peer code reviews, resilient architecture design, and thorough testing methodologies.

• Previous experience in a Staff-level engineering position.

• Skills in technical leadership and mentorship.

• Experience employing AI technologies to improve decision-making, enhance workflows, increase efficiency, and achieve business goals.

• Previous experience in cybersecurity, intelligence, or high-compliance industries is a plus.

• Contributions to open-source data or AI/ML projects are a bonus.

🏝️ Benefits

• Competitive compensation and equity awards that lead the market.

• Comprehensive programs for physical and mental wellness.

• Generous vacation and holiday policies to ensure time for rest and rejuvenation.

• Paid parental and adoption leave.

• Opportunities for professional development available to all employees, regardless of their level or role.

• Employee Networks, local community groups, and volunteering opportunities to foster connections.

• Dynamic office culture featuring world-class amenities.

• Health insurance coverage.

• 401k plan.

• Paid time off.

• Eligibility for bonuses.

• Equity grants.