Research Engineer / Scientist - AI for Databases

1 Day ago • All levels • Research & Development • Artificial Intelligence • $145,000 PA - $355,000 PA

Job Summary

Job Description

ByteDance's Infrastructure System Lab seeks a Research Engineer/Scientist to focus on AI for Databases. This role involves designing and implementing intelligent systems to enhance database performance, scalability, and usability. Responsibilities include developing AI/ML algorithms for query planning, indexing, storage, and workload prediction; integrating AI models into production systems; analyzing large datasets; and publishing research findings in top-tier conferences. The ideal candidate will possess a PhD in a relevant field, a strong publication record in AI4DB, and experience with database internals and machine learning frameworks. Collaboration with engineering teams and contribution to open-source projects are key aspects of this position.
Must have:
  • PhD in CS/Data Science (AI/DB focus)
  • Strong publication record (SIGMOD, VLDB, etc.)
  • Database internals expertise
  • ML framework experience (TensorFlow, PyTorch)
  • Develop intelligent DB algorithms
  • Integrate AI models into production
Good to have:
  • Proficiency in Python/C++/Java
  • Cloud database platform experience
  • LLM, reinforcement learning, or automated DB tuning familiarity
Perks:
  • Competitive compensation
  • Strong research support
  • Innovation-driven environment
  • Medical, dental, and vision insurance
  • 401k savings plan
  • Paid parental leave
  • Paid time off

Job Details

Responsibilities
About the Team The Infrastructure System Lab is a hybrid research and engineering team dedicated to building the next generation of AI-native data infrastructure. Operating at the crossroads of databases, large-scale systems, and AI, the team innovates across multiple domains, including advanced VectorDBs and multi-modal databases for large-scale retrieval and reasoning, intelligent infrastructure optimization using machine learning, LLM-based developer tools like NL2SQL and NL2Chart, and high-performance cache systems for distributed storage and LLM inference. The lab is deeply collaborative, with researchers and engineers working side by side to turn groundbreaking ideas into production-ready systems. Their work is deployed at scale, powering products used by millions, and frequently shared through publications and open-source contributions. About the Role We are looking for a passionate and skilled professional to join our AI4DB team, where artificial intelligence meets cutting-edge database technology. In this role, you will design and implement intelligent systems that improve the performance, scalability, and usability of modern databases. Your work will span query optimization, indexing strategies, workload forecasting, and the development of self-managing databases. This role offers the chance to solve complex, high-impact problems at the intersection of AI, systems, and software engineering, while collaborating with a top-tier team. You'll have opportunities to publish, contribute to open-source, attend leading conferences, and benefit from competitive compensation, strong research support, and an innovation-driven environment. Responsibilities - Conduct research and development in applying AI/ML techniques to database management systems. - Develop intelligent algorithms for tasks such as query planning, indexing, storage management, and workload prediction/scheduling. - Collaborate with data infrastructure and engineering teams to integrate AI models into production systems. - Analyze large-scale datasets from database workloads to uncover optimization opportunities. - Publish findings in top-tier conferences and journals (VLDB, SIGMOD, ICDE, NeurIPS, etc.). - Contribute to open-source projects or internal tools supporting AI-enhanced databases.
Qualifications
Minimum Qualifications - PhD in Computer Science, Data Science or a related field with a focus on databases, systems, or machine learning. - Strong publication record in top-tier venues (e.g., SIGMOD, VLDB, ICDE, NeurIPS, etc.) related to the AI4DB area. - Strong background in database internals (e.g., PostgreSQL, MySQL, or any modern cloud-native databases or BigData platform). - Hands-on experience with machine learning frameworks (e.g. XGBoost, LightGBM, TensorFlow, PyTorch, scikit-learn). Preferred Qualifications - Proficiency in Python, C++, or Java. - Experience with cloud database platforms (AWS, GCP, Azure) is a plus. - Strong analytical, problem-solving, and communication skills. - Familiarity with LLM, reinforcement learning, neural architecture search, or automated database tuning.
Job Information
【For Pay Transparency】Compensation Description (Annually)

The base salary range for this position in the selected city is $145000 - $355000 annually.

Compensation may vary outside of this range depending on a number of factors, including a candidate’s qualifications, skills, competencies and experience, and location. Base pay is one part of the Total Package that is provided to compensate and recognize employees for their work, and this role may be eligible for additional discretionary bonuses/incentives, and restricted stock units.

Benefits may vary depending on the nature of employment and the country work location. Employees have day one access to medical, dental, and vision insurance, a 401(k) savings plan with company match, paid parental leave, short-term and long-term disability coverage, life insurance, wellbeing benefits, among others. Employees also receive 10 paid holidays per year, 10 paid sick days per year and 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure).

The Company reserves the right to modify or change these benefits programs at any time, with or without notice.

For Los Angeles County (unincorporated) Candidates:

Qualified applicants with arrest or conviction records will be considered for employment in accordance with all federal, state, and local laws including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act. Our company believes that criminal history may have a direct, adverse and negative relationship on the following job duties, potentially resulting in the withdrawal of the conditional offer of employment:

1. Interacting and occasionally having unsupervised contact with internal/external clients and/or colleagues;

2. Appropriately handling and managing confidential information including proprietary and trade secret information and access to information technology systems; and

3. Exercising sound judgment.

About Us

Founded in 2012, ByteDance's mission is to inspire creativity and enrich life. With a suite of more than a dozen products, including TikTok, Lemon8, CapCut and Pico as well as platforms specific to the China market, including Toutiao, Douyin, and Xigua, ByteDance has made it easier and more fun for people to connect with, consume, and create content.

Why Join ByteDance

Inspiring creativity is at the core of ByteDance's mission. Our innovative products are built to help people authentically express themselves, discover and connect – and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and enrich life - a mission we work towards every day.

As ByteDancers, we strive to do great things with great people. We lead with curiosity, humility, and a desire to make impact in a rapidly growing tech company. By constantly iterating and fostering an "Always Day 1" mindset, we achieve meaningful breakthroughs for ourselves, our Company, and our users. When we create and grow together, the possibilities are limitless. Join us.

Diversity & Inclusion

ByteDance is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At ByteDance, our mission is to inspire creativity and enrich life. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.

Reasonable Accommodation

ByteDance is committed to providing reasonable accommodations in our recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs or other reasons protected by applicable laws. If you need assistance or a reasonable accommodation, please reach out to us at

Similar Jobs

Next Level Business Services - SAP Security & GRC Admin

Next Level Business Services

Charlotte, North Carolina, United States (On-Site)
6 Months ago
Google - Senior Software Engineer, Full Stack, Google Cloud Technical Infrastructure

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
ByteDance - Senior Software Engineer - Development Infrastructure Team

ByteDance

Mountain View, California, United States (On-Site)
5 Months ago
ByteDance - Senior Software Engineer, Backend and Infrastructure

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Netflix - Data Engineer (L4) - Security

Netflix

United States (Hybrid)
2 Months ago
Starkflow - Specialist

Starkflow

Bengaluru, Karnataka, India (On-Site)
1 Month ago
ByteDance - Interaction Technology Lead - Smart Wearable Devices- Pico Lab- San Jose

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
ByteDance - Imaging Systems Lead - Smart Wearable

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
Playtika - Java Technical Lead

Playtika

Romania (Hybrid)
5 Months ago
Google - Software Engineering Manager II

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Trend Micro - Cloud Engineer (Golang/Python, Backend Focus) 雲端開發工程師

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
Google - Software Engineering Manager, Cloud Observability Analytics

Google

Pittsburgh, Pennsylvania, United States (On-Site)
1 Day ago
Microsoft - Software Engineer 2

Microsoft

Hyderabad, Telangana, India (On-Site)
1 Week ago
Saviynt - Technical Lead, Field Engineering

Saviynt

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
Google - Senior Software Engineer, Full Stack, Google Cloud

Google

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Axinous - Principal Software Development Engineer

Axinous

(Remote)
2 Months ago
Google - Silicon Software Test Engineer

Google

Bengaluru, Karnataka, India (On-Site)
1 Week ago
ByteDance - Senior Software Engineer, Multi Cloud CDN

ByteDance

Seattle, Washington, United States (On-Site)
1 Day ago
NVIDIA - Senior Firmware Bringup Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
2 Weeks ago
Playrix - Director of Engineering

Playrix

Georgia (Remote)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in San Jose, California, United States

Survios - Gameplay Animator

Survios

California, United States (Remote)
2 Months ago
Google - Technical Program Manager II, Network Capacity Planning, Cloud Infrastructure

Google

Austin, Texas, United States (On-Site)
1 Day ago
Zones - Client Solutions Architect

Zones

Baltimore, Maryland, United States (Remote)
1 Month ago
ION - Senior Business Consultant - Endur

ION

Houston, Texas, United States (On-Site)
6 Months ago
ByteDance - Senior Optical System Engineer

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
SKYDANCE - Assistant to Producer and Director

SKYDANCE

Santa Monica, California, United States (Hybrid)
4 Weeks ago
Meta - Software Engineer, iOS

Meta

San Francisco, California, United States (On-Site)
5 Months ago
PlayStation Global - Global Entertainment Services Manager (Latin America)

PlayStation Global

Los Angeles, California, United States (On-Site)
2 Months ago
Next Level Business Services - CCI News Gate Solution Architect

Next Level Business Services

Jersey City, New Jersey, United States (On-Site)
6 Months ago
ION - Services Director

ION

New York, New York, United States (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Meta - ASIC Engineer, Design Verification

Meta

Sunnyvale, California, United States (Remote)
5 Months ago
Fluence - Controls Software Engineer-II(m/f/d)

Fluence

Berlin, Berlin, Germany (Hybrid)
6 Months ago
Krafton  - HR Planning Specialist (5+ years)

Krafton

Seoul, South Korea (On-Site)
2 Weeks ago
Google - Research Scientist, Paradigms of Intelligence

Google

Zürich, Zurich, Switzerland (On-Site)
1 Week ago
Krafton  - [PUBG IP Franchise] PUBG Mobile Global PM (7년 이상)

Krafton

Seoul, South Korea (On-Site)
4 Months ago
NVIDIA - Senior Architect, NVLink

NVIDIA

Westford, Massachusetts, United States (On-Site)
1 Month ago
NVIDIA - Senior Math Libraries Engineer – AI and HPC

NVIDIA

Santa Clara, California, United States (Remote)
2 Months ago
Google - Senior Battery Design Mechanical Engineer

Google

Mountain View, California, United States (On-Site)
1 Week ago
ByteDance - Software Engineer, Model Inference

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
Google - Senior CPU RTL Designer

Google

Mountain View, California, United States (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

About The Company

Where imagination meets innovation, delivering limitless gaming experiences.

San Diego, California, United States (On-Site)

San Jose, California, United States (On-Site)

Dubai, Dubai, United Arab Emirates (On-Site)

New York, New York, United States (On-Site)

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

Seattle, Washington, United States (On-Site)

View All Jobs

Get notified when new jobs are added by ByteDance

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug