TensorFlow

TensorFlow Developer Jobs

21 jobs found

Today

Principal Machine Learning Engineer, Accelerated Apache Spark

NVIDIA Santa Clara, CA

Santa Clara, CA

Today

$272k - $431k/yr Full time Python Big Data Bigdata

Apply Now

Job Description

NVIDIA offers equity and a benefits package, with an on-site机会 in Santa Clara, CA. This Principal Machine Learning Engineer role sits on the Accelerated Apache Spark team, focusing on ML driven performance prediction and optimization for GPU-accelerated Spark workloads. You will lead ML engineering efforts, mentor engineers, and help deploy AI based tooling across multiple environments, all within a culture that values technical excellence and collaboration. The position carries a salary range of USD 272,000 to 431,250 per year and requires onsite presence.

Responsibilities

Design and implement machine learning solutions for predicting and optimizing performance of GPU-accelerated enterprise Apache Spark workloads.
Develop advanced algorithms and adaptive systems to continually improve Spark performance on GPUs.
Build AI based agents and tools to assist with diagnosing issues and optimizing applications.
Partner with key stakeholders and customers to deploy complex ML solutions across diverse environments.
Maintain deep domain expertise by tracking the latest advances in ML systems and algorithms.
Provide technical mentorship and leadership in data science and machine learning for a team of engineers.

Requirements

BS, MS, or PhD or equivalent experience in Machine Learning, Data Science, Computer Science or a closely related field.
12+ years of professional experience designing, implementing, and productionizing high quality ML/DL solutions.
5+ years as a technical lead in ML model development.
Proven hands-on experience (2+ years) with large-scale data processing platforms such as Apache Spark.
Proven ability to employ modern tooling and sound techniques for crafting, deploying, and maintaining ML models.
Excellent programming skills in Python and related data science libraries (numpy, pandas, scikit-learn, scipy, pytorch, tensorflow).
Deep experience with ML methodologies including LLM/GenAI, reinforcement learning, and adaptive online ML systems.
Strong expertise in feature engineering, feature importance assessment, and developing boosted tree models (e.g., XGBoost).

Technologies

Python
numpy
pandas
scikit-learn
scipy
pytorch
tensorflow
Apache Spark
XGBoost
Scala
Java
C++
CUDA

Benefits

Equity and benefits

Ways to Stand Out

Understanding of the internal workings and architecture related to Apache Spark
Familiarity with NVIDIA GPUs and CUDA
Experience coding in Scala, Java, and/or C++

View full job page →

Today

Senior Machine Learning Engineer, End‑to‑End Autonomous Driving

NVIDIA Santa Clara, CA

Santa Clara, CA

Today

$184k - $357k/yr Full time Senior Active Learning Artificial Intelligence

Apply Now

Job Description

Join NVIDIA in Santa Clara, onsite, to lead the design, training, and deployment of end-to-end autonomous driving models. This senior role offers a salary range of USD 184,000 to 356,500 per year, plus equity and a comprehensive benefits package. You will collaborate across teams to build data-centric pipelines, drive data flywheels, and translate research into robust, production-grade machine learning systems for autonomous driving.

Benefits

Equity
Benefits

Responsibilities

Design, implement, and train large-scale end-to-end driving models.
Lead the data flywheel by identifying failure cases, specifying data collection and labeling needs, and iterating models to close real-world gaps in performance.
Build, curate, and maintain high-quality multimodal datasets (video, sensor, language/action traces) tailored for end-to-end autonomous driving.
Apply data-centric learning methods such as active learning, curriculum learning, automated hard-example mining, outlier and novelty detection, and semi/self-supervised approaches.
Explore and productize new data sources including simulation, synthetic data, and world-model based generation to improve coverage and robustness.
Design data workflows that automate data discovery, labeling, evaluation, and retraining to maximize development velocity.
Foster collaborative partnerships with researchers and engineers to transform innovative research into robust, industrial-strength models.

Requirements

PhD with 4+ years, MS with 6+ years, or BS (or equivalent experience) with 8+ years of relevant work in Computer Science, Computer Engineering, or a related technical field.
Strong background in modern deep learning, including transformer-based architectures, video modeling, and multimodal VLM/VLA or foundation models.
Hands-on experience training and deploying deep learning models on real-world datasets: data preprocessing, distributed training, evaluation, debugging, and iterative improvement.
Practical experience with data-centric methods such as active learning, curriculum learning, outlier/novelty detection, or large-scale sample mining.
Proficiency in Python and at least one major deep learning framework (PyTorch, TensorFlow, or JAX), along with solid software engineering practices (testing, code review, CI/CD).
Proven ability to collaborate across teams, drive designs from prototype to production, and communicate clearly with technical and non-technical partners.
A track record of leading complex cross-team projects, setting technical direction, and making critical decisions that impact multiple teams or products.

Technologies

Python
PyTorch
TensorFlow
JAX

Ways to stand out

Experience building and operating data flywheels or large-scale ML data pipelines, including data quality monitoring and continuous retraining loops.
Direct experience with end-to-end driving models, large-scale behavior cloning, or reinforcement/imitation learning for driving or robotics.
Experience leveraging simulation, synthetic data, or world models to generate training and evaluation data for autonomous systems.
Contributions to advanced data-centric ML methods, VLM/VLA, or autonomous driving through publications, open-source projects, or widely used internal tools.
Background with safety, reliability, and validation requirements for autonomous driving or other safety-critical applications.

View full job page →

4d ago

Machine Learning Engineer, Ads Creative

TikTok San Jose, CA

San Jose, CA

4d ago

$156k - $317k/yr Full time Ad Tech Ai Ml Deep Learning

Apply Now

Job Description

TikTok is seeking a Machine Learning Engineer for Ads Creative to help design and scale large-scale ads systems, applying cutting-edge ML to monetization, targeting, and creative optimization. This role sits at the intersection of advanced algorithm development and product impact, collaborating with product and business teams to align ML initiatives with a clear product vision. Based in San Jose and onsite, you will contribute to the evolution of TikTok’s advertising ecosystem.

Responsibilities

Participate in the development of a large-scale Ads system.
Develop state-of-the-art applied machine learning projects.
Own key targeting components or strategies in the TikTok ads monetization ecosystem.
Collaborate with product and business teams on product vision.

Requirements

BS/MS degree in Computer Science, Computer Engineering, or other relevant majors.
Experience in and strong theoretical grounding in machine learning concepts and techniques.
Excellent programming, debugging, and optimization skills in one or more general purpose programming languages, including but not limited to: Go, C/C++, Python.
Experience in one or more frameworks: Tensorflow/PyTorch/MXNet, etc.
Ability to think critically and formulate solutions to problems in a clear and concise way.

Education

BS/MS degree in Computer Science or Computer Engineering, or other relevant majors.

Technologies

Go
C
C++
Python
Tensorflow
PyTorch
MXNet

Compensation

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

Benefits

Medical, dental, and vision insurance
401(k) savings plan with company match
Paid parental leave
Short-term and long-term disability coverage
Life insurance
Wellbeing benefits
10 paid holidays per year
10 paid sick days per year
17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure)
Potential discretionary bonuses/incentives
Restricted stock units

Location

San Jose, CA (onsite)

Employment Type

Regular

About TikTok

TikTok is the leading destination for short-form mobile video. Our mission is to inspire creativity and bring joy. TikTok's global headquarters are in Los Angeles and Singapore, with offices worldwide.

Why Join Us

Inspiring creativity is at the core of TikTok's mission. Our innovative product helps people authentically express themselves, discover, and connect, powered by diverse teams that make that possible. We work to create value for communities, inspire creativity, and bring joy every day.

Diversity & Inclusion

TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people globally, and our workplace reflects the many communities we reach. We are dedicated to celebrating diverse voices and fostering an environment that mirrors our global reach.

TikTok Accommodation

TikTok provides reasonable accommodations in our recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs, or other protected reasons. If you need assistance or a reasonable accommodation, please reach out at https://tinyurl.com/RA-request

View full job page →

7d ago

Lead Generative AI Data Engineer III

Deloitte Philadelphia, PA

Philadelphia, PA

7d ago

$159k - $265k/yr Full time Ai Ml Ai Workflows Anthropic

Apply Now

Job Description

Deloitte seeks a Lead Generative AI Data Engineer III to head the end-to-end lifecycle of ML and AI initiatives, onsite in Philadelphia.

Responsibilities

Lead the design, development, testing, and deployment of machine learning and artificial intelligence solutions for business and client use cases.
Steer AI engineering workstreams by assigning tasks, reviewing outputs, and driving quality, timeliness, and issue resolution across the team.
Build and supervise data pipelines, model training workflows, and production-grade components that underpin AI enabled products.
Collaborate with engineers, data scientists, product stakeholders, and business teams to translate requirements into scalable technical solutions and guide key technical decisions.
Coach and mentor junior practitioners, provide daily guidance, and monitor model and application performance to improve accuracy, reliability, and scalability.

Requirements

Bachelor's degree in Computer Science, Data Science, Engineering, Mathematics, Statistics, or a related quantitative field.
7+ years of professional experience.
5+ years designing, developing, or deploying machine learning, artificial intelligence, or advanced analytics solutions.
3+ years programming in Python, PySpark, PyTorch, and TensorFlow.
2+ years of technology consulting experience in the Federal Health sector.
2+ years deploying machine learning models into production environments.
2+ years leading technical teams, projects, or workstreams delivering AI or ML solutions.
1+ years working with AI technologies such as Gemini, Anthropic, and OpenAI.
Active certification or advanced certification in Python, PySpark, PyTorch, and TensorFlow.
Ability to travel approximately 20% based on client engagements.
Authorized to work in the United States without employer sponsorship now or in the future.

Technologies

Python
PySpark
PyTorch
TensorFlow
Docker
Kubernetes
Gemini
Anthropic
OpenAI

Benefits

Discretionary annual incentive program

The Team

Deloitte's Government & Public Services GPS practice delivers impact through people, ideas, technology, and outcomes for federal, state, and local government clients as well as public higher education institutions. The AI & Data offering provides end-to-end solutions for designing, developing, and operating data and AI platforms, products, insights, and services to help clients mature and scale their AI capabilities with strategic programs.

Compensation

The listed salary range reflects multiple factors including skills, experience, training, certifications, and business needs. Geographic differences may apply, and compensation is not guaranteed at the top of the range. Eligible employees may participate in a discretionary annual incentive program based on performance and company factors.

View full job page →

8d ago

Machine Learning Engineer

Booz Allen Hamilton Arlington, VA

Arlington, VA

8d ago

$78k - $176k/yr Full time Ai Ml Artificial Intelligence Data Science

Apply Now

Job Description

Machine Learning Engineer position at Booz Allen Hamilton focusing on training, deploying, and maintaining ML models and guiding AI strategy across the DoD market, collaborating across engineering and product teams.

Responsibilities

Develop, evaluate, deploy, and sustain models that learn from data.
Define the direction of mission critical AI solutions to enable faster mission success.
Join a large community of ML engineers across the company and collaborate with data engineers, data scientists, solutions architects, and product owners to deliver advanced AI solutions across diverse use cases.
Apply deep technical expertise to guide clients through ML algorithms, tools, and frameworks.

Requirements

Minimum 2 years of experience in artificial intelligence, data science, or machine learning engineering, including developing and deploying models.
Proficiency in Python and libraries such as scikit-learn, TensorFlow, or PyTorch.
Experience coding in a common programming language, including Python, C++, Rust, or Java.
Experience with deep learning, computer vision, or generative AI.
Experience with data handling and model development lifecycle.
Experience leveraging AI productivity tools for coding, such as Cursor.
Ability to solve client issues and problems.
Secret security clearance.
Bachelor's degree.

Nice to Have

Experience with generative and agentic AI technologies, such as LLMs, MCP, LangChain, or LangGraph.
Experience with supervised and unsupervised learning.

Clearance

Applicants selected will be subject to a security investigation and may need to meet eligibility requirements for access to classified information; Secret clearance is required.

Technologies

Python
scikit-learn
TensorFlow
PyTorch
C++
Rust
Java
Cursor

Location

Arlington, VA onsite

Compensation

Salary range: USD 77,600 - 176,000 per year.

Benefits

Health benefits
Life insurance
Disability benefits
Financial benefits
Retirement benefits
Paid leave
Professional development
Tuition assistance
Work-life programs
Dependent care

Identity and Compliance

Identity verification process leveraging advanced biometrics and artificial intelligence during hiring.
Expect to be on camera during interviews and assessments.

AI Usage Policy

AI use for responses during interviews is prohibited unless explicit permission is provided.

Work Model

Onsite: work primarily at Booz Allen office or customer facility, with collaboration as required.

Non-Discrimination

All qualified applicants will receive consideration regardless of disability, protected veteran status, or any other protected status under applicable law.

View full job page →

11d ago

Machine Learning Engineer

TikTok San Jose, CA

San Jose, CA

11d ago

$156k - $388k/yr Full time Ai Ml Artificial Intelligence Deep Learning

Apply Now

Job Description

Machine Learning Engineer at TikTok focused on multimodal content understanding and short-video recommendation, leading algorithm research, development, and deployment to improve business metrics.

Responsibilities

Head the development of multimodal algorithms powering TikTok's short-form video business; explore applications of multimodal technologies in recommendations and other contexts to improve key metrics.
Pursue cutting-edge research in multimodal and multimodal large language model technologies; craft advanced algorithms that meet business needs while pushing technical boundaries.
Lead deployment and integration efforts, ensuring model stability, scalability, and efficiency in production environments.
Contribute to AI platform design and development, including few-shot/zero-shot on MLLMs, AI labeling, auto prompting, active learning, continued pretraining, and reinforcement learning.
Bridge content understanding with recommendation systems across UGC ecosystems, addressing cold start, interest exploration, and comment understanding.
Apply multimodal techniques to build next-generation recommendation systems, including generative models and end-to-end approaches.

Requirements

Demonstrated experience in multimodal content understanding, with strong expertise in large language models and familiarity with recent advances in the field.
Solid technical foundation in at least one major deep learning framework (for example PyTorch or TensorFlow).
Self-motivated with a strong ownership mindset, excellent communication, and ability to collaborate across teams.
Hands-on experience deploying content understanding solutions in search, advertising, recommendation, or related domains.

Technologies

PyTorch
TensorFlow
LLMs
MLLM

Benefits

Medical, dental, and vision insurance from day one.
401(k) savings plan with company match.
Paid parental leave.
Short-term and long-term disability coverage.
Life insurance.
Wellbeing benefits.
10 paid holidays per year.
10 paid sick days per year.
17 days of Paid Personal Time, prorated at hire with tenure-based accrual increases.

Job Information

Location: San Jose, CA (onsite)
Base salary: USD 156,000 - 387,600 per year
Compensation may vary outside this range based on qualifications, skills, competencies, experience, and location; base pay is part of the Total Package and may include discretionary bonuses/incentives and restricted stock units.
Benefits may vary by employment type and country; day-one access to medical, dental, and vision, 401(k) with match, paid parental leave, disability coverage, life insurance, wellbeing benefits, holidays, sick days, and paid personal time.
TikTok reserves the right to modify or change benefits programs at any time, with or without notice.

LA County Candidates

Interacting and occasionally having unsupervised contact with internal/external clients and colleagues.
Handling and managing confidential information, including proprietary and trade secret data, and access to IT systems.
Exercising sound judgment.

About TikTok

TikTok is the leading destination for short-form mobile video. Our mission is to inspire creativity and bring joy. Global headquarters are in Los Angeles and Singapore, with offices worldwide.

Why Join Us

Inspiring creativity lies at the heart of TikTok's mission. Our product enables authentic self-expression, discovery, and connection, supported by diverse, global teams.

Diversity & Inclusion

TikTok is committed to an inclusive workplace that values diverse skills, experiences, and perspectives. We celebrate diverse voices and strive to reflect the communities we reach.

TikTok Accommodation

TikTok provides reasonable accommodations in recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs, or other protected reasons. For assistance, please submit a RA request at https://tinyurl.com/RA-request

View full job page →

11d ago

Senior Machine Learning Engineer

TikTok San Jose, CA

San Jose, CA

11d ago

$156k - $388k/yr Full time Senior Deep Learning Engineering

Apply Now

Job Description

Senior Machine Learning Engineer on TikTok's Recommendation team, building and deploying end-to-end ML systems to strengthen the US market.

Responsibilities

Design and refine recommendation models that enhance user experience and shape the platform’s content ecosystem.
Construct ML systems and data pipelines that address key product challenges at scale.
Own end-to-end ML stacks, guiding development, maintenance, and ongoing improvements.
Collaborate with cross functional teams to shape product strategy and deliver US market growth initiatives.

Requirements

Bachelor degree or higher in computer science or a related field with at least 4 years of relevant experience.
Strong proficiency in C/C++ and Python, with solid data structures and algorithms knowledge and hands-on experience building large-scale production systems.
Practical background developing ML systems using at least one major deep learning framework such as PyTorch or TensorFlow.
Experience in applied ML, ML infrastructure, large-scale recommendation systems, or market-facing ML products.
Effective written and verbal communication skills.

Technologies

C
C++
Python
PyTorch
TensorFlow

Compensation

Annual base salary: $156,000 – $387,600
Compensation may vary based on qualifications, skills, experience, and location. Total package may include discretionary bonuses/incentives and restricted stock units.

Benefits

Medical, dental, and vision insurance
401(k) savings plan with company match
Paid parental leave
Short-term and long-term disability coverage
Life insurance
Wellbeing benefits
Paid holidays, sick days, and personal time

About the Team

The recommendation algorithm team drives critical product decisions and platform growth.
Researchers and engineers from diverse backgrounds collaborate to support production recommendation systems with real impact.
The team operates in a fast-paced, collaborative, and impact-driven environment.

General Hire

If applying under the General Hire track for the TikTok Recommendation Organization, you will undergo a unified interview process to assess fit across the broader organization, with placement determined by background, interests, and interview performance.

Location

Location: San Jose, CA onsite

Accommodations

TikTok provides reasonable accommodations in recruitment processes for candidates with disabilities or other protected needs. To request an accommodation, contact us at RA request.

About TikTok

TikTok is a leading destination for short-form mobile video with a mission to inspire creativity and bring joy.
Global headquarters are in Los Angeles and Singapore, with offices worldwide including New York, London, Dublin, Paris, Berlin, Dubai, Jakarta, Seoul, and Tokyo.

Why Join Us

Be part of a mission that centers on creativity, discovery, and connection through innovative products.
Work with curious, humble colleagues to drive meaningful impact in a fast-growing tech company.
Collaborate across teams to achieve breakthroughs while maintaining an Always Day 1 mindset.

Diversity & Inclusion

TikTok values diverse voices and perspectives, creating an inclusive space for all employees.
We aim to reflect the global communities we reach in our workforce and culture.