ByteDance

Research Engineer / Scientist -AI for Databases

ByteDance
Apply
3 hours ago
San Jose, CA, USAEntry Level
H1B sponsor

Responsibilities

  • Conduct research and development applying AI/ML techniques to database management systems.
  • Develop intelligent algorithms for query planning, indexing, storage management, workload prediction, and scheduling.
  • Collaborate with data infrastructure and engineering teams to integrate AI models into production systems.
  • Analyze large-scale database workload datasets to identify optimization opportunities.
  • Publish findings in conferences and journals including VLDB, SIGMOD, ICDE, and NeurIPS.
  • Contribute to open-source projects or internal tools supporting AI-enhanced databases.

Requirements

  • PhD in Computer Science, Data Science, or a related field focused on databases, systems, or machine learning.
  • Strong publication record in accredited AI4DB-related venues such as SIGMOD, VLDB, ICDE, or NeurIPS.
  • Strong database internals background, including PostgreSQL, MySQL, modern cloud-native databases, or BigData platforms.
  • Hands-on experience with machine learning frameworks such as XGBoost, LightGBM, TensorFlow, PyTorch, or scikit-learn.
  • Preferred: proficiency in Python, C++, or Java.
  • Preferred: experience with AWS, GCP, or Azure cloud database platforms.
  • Preferred: familiarity with LLMs, reinforcement learning, neural architecture search, or automated database tuning.
  • Strong analytical, problem-solving, and communication skills.

Benefits

  • Opportunities to publish research, attend leading conferences, and contribute to open-source projects.
  • Strong research support and an innovation-driven environment.
  • Work on systems deployed at scale for products used by millions.
  • Competitive compensation is mentioned, but no amount is specified.

Tech Stack

AWSAzureC++Google Cloud PlatformJavaLightGBMMySQLPostgreSQLPythonPyTorchscikit-learnTensorFlowXGBoost
ByteDance

About ByteDance

10,000+ employees

ByteDance is a global incubator of platforms at the cutting edge of commerce, content, entertainment and enterprise services - over 2.5bn people interact with ByteDance products including TikTok. Creation is the core of ByteDance's purpose. Our products are built to help imaginations thrive. This is doubly true of the teams that make our innovations possible. Together, we inspire creativity and enrich life - a mission we aim towards achieving every day. At ByteDance, we create together and grow together. That's how we drive impact - for ourselves, our company, and the users we serve. We are committed to building a safe, healthy and positive online environment for all our users. We have over 110,000 employees based in more than 30 countries globally. Join us.

Contact me