Drimva

ByteDance

Research Engineer / Scientist -AI for Databases

San Jose, California · Full-time · Posted Oct 28, 2026

Education
PhD in Computer Science, Data Science, or a related field with a focus on databases, systems, or machine learning.

About the job

Research and develop AI/ML-powered database systems that improve query planning, indexing, storage management, workload prediction, and database performance.

What you'll do

  • Develop intelligent algorithms for database management and optimization.
  • Integrate AI models into production systems with infrastructure and engineering teams; analyze database workload data for optimization opportunities.
  • Publish research and contribute to open-source projects or internal tools for AI-enhanced databases.

What they ask for

  • PhD in Computer Science, Data Science, or a related field, focused on databases, systems, or machine learning.
  • Strong publication record in accredited venues related to AI4DB.
  • Strong background in database internals.
  • Hands-on experience with machine learning frameworks.

Nice to have

  • Proficiency in Python, C++, or Java; experience with AWS, GCP, or Azure database platforms.
  • Familiarity with LLM, reinforcement learning, neural architecture search, or automated database tuning.

Tools

PostgreSQL, MySQL, XGBoost, LightGBM, TensorFlow, PyTorch, scikit-learn, Python

AI focus

Apply AI/ML to database query planning, indexing, storage management, workload prediction and scheduling, and automated database tuning.

Model-written summary · the original posting is authoritative

First seen Oct 28, 2026 · Last verified Nov 5, 2026 (UTC)