PyTorch

PyTorch Developer Jobs

87 jobs found

1d ago

Gen AI Developer

Realign LLC Plano, TX

Plano, TX

1d ago

$105k+/yr Full time AI Artificial Intelligence Cloud Platform

Apply Now

Job Description

Realign LLC is hiring a Gen AI Developer in Plano, TX (onsite) to design, fine-tune, deploy, and support generative AI and LLM solutions.

Responsibilities

Design, fine-tune, and deploy generative AI models for production applications, including Llama, GPT, and Stable Diffusion
Train and optimize large language models (LLMs) for natural language understanding, summarization, and conversational AI
Build and maintain data pipelines that support both model training and inference
Clean, preprocess, and manage large-scale datasets used across AI initiatives
Implement machine learning models in production using TensorFlow, PyTorch, or Hugging Face
Optimize deployed models for performance, scalability, and cost-efficiency on AWS, Azure, and GCP
Collaborate with product managers, data engineers, and software developers to connect AI deliverables to business objectives
Track and apply advancements in Gen AI and ML research
Monitor deployed models for performance and accuracy; execute retraining and versioning strategies
Ensure AI systems meet ethical, privacy, and compliance standards

Requirements

Experience in machine learning, data science, or a related discipline
Hands-on experience with generative models such as GANs, VAEs, and diffusion models, plus LLMs
Experience with ML frameworks including TensorFlow, PyTorch, or JAX
Familiarity with cloud ML environments such as AWS Sagemaker, Azure ML, or GCP AI Platform
Strong programming skills in Python (preferred) or similar languages
Proficiency with tools and libraries such as Hugging Face, LangChain, or NVIDIA BioNeMo
Knowledge of Docker, Kubernetes, and CI/CD pipelines for deployment
Understanding of NLP, computer vision, or multimodal AI techniques
Strong problem-solving skills and interest in AI-driven innovation
Experience with Retrieval-Augmented Generation (RAG) and vector databases such as Pinecone, Weaviate, or Milvus

Salary: USD 105,000 per year

Technologies: machine learning, data science, generative models, GANs, VAEs, diffusion models, LLMs, TensorFlow, PyTorch, JAX, AWS Sagemaker, Azure ML, GCP AI Platform, Python, Hugging Face, LangChain, NVIDIA BioNeMo, Docker, Kubernetes, CI/CD pipelines, Pinecone, Weaviate, Milvus, Retrieval-Augmented Generation (RAG), Llama, GPT, Stable Diffusion, NLP, computer vision

View full job page →

1d ago

Machine Learning Engineer Graduate (E-Commerce User Growth)

TikTok Seattle, WA

Seattle, WA

1d ago

$122k - $243k/yr Full time Marketing Big Data Data Pipeline

Apply Now

Job Description

Join TikTok in Seattle for a 2027 start graduate role focused on machine learning for e-Commerce user growth and marketing.

Responsibilities

Optimize TikTok e-Commerce growth and marketing algorithms, including user value modelling, personalized messaging, recommendations, and intelligent marketing.
Build a user lifecycle data and value system to address TikTok Mall user growth pain points and business challenges.
Implement product and technical solutions for user growth engines, including personalized push notifications and emails, recommendation handling, and related components to increase e-Commerce DAU and penetration rates.
Improve recommendation quality for new users by optimizing recall and sorting, including traffic source handling relevance and the accuracy and diversity of new user recommendations.
Advance intelligent marketing algorithms using uplift models and operations research methods to improve marketing efficiency and drive e-Commerce GMV growth.

Requirements

Currently completing or recently completed a Bachelor’s degree in Software Development, Computer Science, Computer Engineering, or a related discipline.
Experience with C++ and Python; at least one Big Data tool such as Hive SQL, Spark, or MapReduce.
At least one Deep Learning tool such as TensorFlow or PyTorch.
Strong foundation in algorithms and data structures with excellent coding skills.
Strong learning ability, curiosity, and good communication and teamwork skills.
Also requires/considers individuals completing or recently completing a Master’s degree in Software Development, Computer Science, Computer Engineering, or a related discipline.
Published papers in top AI conferences or journals, or notable results in ACM/Machine Learning competitions.
Experience in recommendation systems, advertising, user growth, intelligent marketing, or related areas is a plus, including expertise in LTV estimation, uplift modelling, operations research, sequence modelling, or multi-scenario modelling optimization.

Technologies

C++, Python
Hive SQL, Spark, MapReduce
TensorFlow, PyTorch
Uplift models, operations research methods
LTV estimation, sequence modelling, multi-scenario modelling optimization

Benefits

Day one access to 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)
Additional discretionary bonuses/incentives and restricted stock units (role may be eligible)

Team Introduction

The TikTok e-Commerce Recommendation and Marketing Algorithm team delivers algorithm and big data work for e-Commerce innovation projects.
Focus on helping users discover and acquire great products to enrich their lives.
Uses recommendation and search algorithms for item discovery, plus risk control algorithms and intelligent platform governance to detect violations and support a secure shopping experience.
Builds intelligent customer service technologies and large-scale product knowledge graphs to improve transaction process efficiency.
Develops logistics and operations research algorithms to enhance supply chain efficiency and applies AI to help merchants improve operational capabilities.
Mission: make high-quality products easily accessible, enabling everyone to enjoy a better life.

Additional Information

Successful candidates must be able to commit to an onboarding date by the end of the year.
Please include availability and graduation date clearly in your resume.
Applicants can apply to a maximum of two positions and will be considered in the order applied.
Applications are reviewed on a rolling basis; applying early is encouraged.

Pay Transparency

Base salary range (selected city): $121,600 - $243,200 annually.
Compensation may vary based on qualifications, skills, competencies, experience, and location.
Base pay is part of the Total Package.

Los Angeles County (Unincorporated) Candidates

Qualified applicants with arrest or conviction records will be considered in accordance with federal, state, and local laws, including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act.
The company notes criminal history may have a direct, adverse, negative relationship on job duties, potentially leading to withdrawal of a conditional offer.
Areas cited include interacting with clients or colleagues (occasionally unsupervised), managing confidential/proprietary information, and exercising sound judgment.

TikTok Accommodation

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

View full job page →

3d ago

Director, Machine Learning Engineer

Information Technology Senior Management Forum McLean, VA

McLean, VA

3d ago

$269k - $307k/yr Full time Manager Ai Ml Artificial Intelligence

Apply Now

Job Description

In McLean, VA, this Director-level Machine Learning Engineering role focuses on leading large-scale machine learning initiatives for financial services. The position emphasizes building production-ready ML models and the software and data systems that support them, with strong engineering leadership alongside hands-on technical work.

As part of the Information Technology Senior Management Forum, you will collaborate across Product, Architecture, Engineering, and Data Science to deliver intelligent systems that solve complex business problems, while optimizing data pipelines and deploying models reliably at cloud scale.

Responsibilities

Deliver ML models and software components to address challenging needs in the financial services industry in partnership with Product, Architecture, Engineering, and Data Science teams
Drive the creation and ongoing evolution of ML models and software for state-of-the-art intelligent systems
Lead large-scale ML initiatives with a customer-first mindset
Use cloud-based architectures and technologies to deliver optimized ML models at scale
Optimize data pipelines that feed ML models
Apply programming languages such as Python, Scala, or Java to build and enhance systems
Promote best practices across engineering and modeling lifecycles
Recruit, nurture, and retain top engineering talent
Act as a force-multiplier for the team by balancing deep, hands-on contributions and innovation with mentoring and raising the skills of peers and junior engineers
Stay current on technology trends through experimentation, learning new tools and techniques, participating in internal and external technology communities, and mentoring others

Requirements

Bachelor's Degree or higher in Computer Science, Machine Learning, or a related quantitative field (Statistics, Economics, Operations Research, Analytics, Mathematics, Engineering)
At least 3 years of people leadership experience
At least 8 years of experience programming with Python, Java, Golang, or C++
At least 6 years of Machine Learning experience using industry standard frameworks PyTorch or TensorFlow and libraries (Pandas, NumPy, Scikit-learn)
At least 6 years operating large-scale distributed systems (Spark, Ray) to prepare AI or Machine Learning data
At least 5 years deploying and operating production Machine Learning solutions and production services in the cloud (AWS, GCP, Azure) and using Kubernetes for large-scale containerized ML systems

Technologies

Python, Scala, Java, Golang, C++
PyTorch, Tensorflow, Pandas, NumPy, Scikit-learn
Spark, Ray
AWS, GCP, Azure, Kubernetes
Open source frameworks, distributed systems, cloud-based architectures

Benefits

Performance-based incentive compensation, which may include cash bonus(es) and/or long-term incentives (LTI)
Comprehensive, competitive, and inclusive set of health, financial, and other benefits

Preferred Qualifications

Master's or Doctoral Degree in Computer Science, Electrical Engineering, Mathematics, or related field
5+ years of experience managing and leading an engineering team
3+ years of experience architecting and designing resilient, large-scale, production machine learning systems from data preparation through model training and inference
5+ years of experience optimizing ML algorithms, configurations, and infrastructure
5+ years working with Machine Learning techniques (Supervised, semi-supervised, and unsupervised, reinforcement learning, etc.) across model types (Regression, Classification, Clustering, etc.) and model architectures (RNNs, CNNs, LSTMs, Transformers), including training concepts (loss function, hyperparameters, regularization) and evaluating accuracy and diagnosing issues such as underfitting and overfitting
7+ years designing, implementing, and scaling production-ready data pipelines for training and evaluating ML models
Ability to communicate complex technical concepts to a variety of audiences
Driving impacts through conference presentations, papers, blog posts, open source contributions, or patents
Experience hiring and developing high-performing ML engineers with an inspiring leadership style
Highly developed interpersonal, presentation, and communications skills

Location: McLean, VA (onsite)

Compensation: $269,100 - $307,200 per year for Director, Machine Learning Engineer

View full job page →

3d ago

Director, Machine Learning Engineer

Capital One McLean, VA

McLean, VA

3d ago

$269k - $307k/yr Full time Manager Ai Ml Artificial Intelligence

Apply Now

Job Description

Own and scale production machine learning that tackles complex financial services challenges at Capital One. This onsite role in McLean, VA brings together engineering leadership, cloud and distributed systems, and hands-on model delivery to build state-of-the-art intelligent systems. You will lead large-scale ML initiatives, optimize the full lifecycle from data pipelines to inference, and help mentor engineering talent as a force-multiplier.

Compensation: $269,100 - $307,200 per year (McLean, VA). Performance-based incentive compensation may include cash bonus(es) and/or long term incentives (LTI).

What you’ll do

Deliver machine learning models and software components that solve challenging business problems in financial services, partnering with Product, Architecture, Engineering, and Data Science
Drive the creation and evolution of ML models and software for intelligent systems
Lead large-scale ML initiatives with the customer in mind
Leverage cloud-based architectures and technologies to deliver optimized models at scale
Optimize data pipelines that feed ML models
Use programming languages such as Python, Scala, or Java
Evangelize best practices across engineering and modeling lifecycles
Recruit, nurture, and retain top engineering talent, balancing deep hands-on work with mentoring and elevating peers and junior engineers
Stay current with technology trends through experimentation, learning, and participation in internal and external technology communities, including mentoring others

Core qualifications

Bachelor’s Degree or higher in Computer Science, Machine Learning, or a related quantitative field (Statistics, Economics, Operations Research, Analytics, Mathematics, Engineering)
3+ years of people leadership experience
8+ years of experience programming with Python, Java, Golang, or C++
6+ years of machine learning experience using industry-standard frameworks PyTorch or Tensorflow and libraries including Pandas, NumPy, Scikit-learn
6+ years using and operating large-scale distributed systems such as Spark and Ray to prepare AI or machine learning data
5+ years deploying and operating production machine learning solutions and production services in the cloud (AWS, GCP, Azure) and using Kubernetes to manage large-scale containerized ML software

Preferred qualifications

Master’s or Doctoral Degree in Computer Science, Electrical Engineering, Mathematics, or a related field
5+ years managing and leading an engineering team
3+ years architecting and designing resilient, large-scale production machine learning systems from data preparation to model training and inference
5+ years optimizing ML algorithms, configurations, and infrastructure
5+ years with ML techniques (Supervised, semi-supervised, unsupervised, reinforcement learning, etc.), model types (Regression, Classification, Clustering, etc.), and architectures (RNNs, CNNs, LSTMs, Transformers), including training concepts (loss function, hyperparameters, regularization) and diagnosing issues such as underfitting and overfitting
7+ years designing, implementing, and scaling production-ready data pipelines for training and evaluating ML models
Ability to communicate complex technical concepts clearly to a variety of audiences
Experience driving ML impact through conference presentations, papers, blog posts, open source contributions, or patents
Experience hiring and developing high-performing ML engineers with an inspiring leadership style
Highly developed interpersonal, presentation, and communications skills

Benefits

Performance-based incentive compensation, which may include cash bonus(es) and/or long term incentives (LTI)
Comprehensive, competitive, and inclusive benefits supporting your total well-being (eligibility varies by full-time/part-time status, exempt/non-exempt status, and management level)

Technology stack: Python, Scala, Java, Golang, C++, PyTorch, Tensorflow, Pandas, NumPy, Scikit-learn, Spark, Ray, AWS, GCP, Azure, Kubernetes.

Please note that any position posted in Canada is for Capital One Canada, any position posted in the United Kingdom is for Capital One Europe, and any position posted in the Philippines is for Capital One Philippines Service Corp. (COPSSC).

View full job page →

3d ago

Senior / Staff Machine Learning Engineer, Applied AI

Lila Sciences San Francisco, CA

San Francisco, CA

3d ago

$180k - $298k/yr Full time Senior Ai Ml Artificial Intelligence

Apply Now

Job Description

Lila Sciences is seeking a Senior / Staff Machine Learning Engineer, Applied AI to reduce the gap between frontier model capabilities and customer-specific scientific workflows. This role centers on post-training, evaluation loops, model debugging, and integration of model behavior into end-to-end product workflows.

Key Responsibilities

Use hands-on expertise to close the last-mile gap between Lila’s model capabilities and customer-specific scientific workflows.
Lead post-training execution using methods such as SFT and RL, including DPO, PPO/GRPO, to align model behavior with customer requirements and feedback.
Run evaluation loops that assess model quality, reliability, and fit for customer use cases.
Convert customer learnings, data signals, and evaluation outcomes into repeatable model improvement cycles.
Partner with AI researchers to translate model advances into reliable, usable capabilities.
Coordinate with Software to integrate model behavior into end-to-end product workflows.
Debug complex model failures using traces, evaluations, customer context, and scientific feedback.

Required Qualifications

Minimum 1-2 years of hands-on post-training experience, including SFT and RL methods such as DPO/PPO/GRPO, along with evaluation system design developed through repeated, real-world problem solving.
Strong software engineering skills in Python and modern ML frameworks such as PyTorch.
Proven ability to debug ambiguous, high-stakes model behavior quickly using data, traces, logs, and qualitative feedback.
Deep familiarity with large language models, multi-modal models, or agentic AI systems.
Clear communication skills for translating customer needs into technical approaches and explaining complex model behavior to technical and non-technical audiences.

Technologies

Python
PyTorch
SFT
RL
DPO
PPO
GRPO

Bonus Points

Experience adapting models for customer-facing or production workflows, preferably in scientific, technical, or data-intensive domains.
Experience with RL post-training such as RLHF, GRPO, or tool-augmented RL.
Experience building evaluation harnesses, model monitoring, or quality dashboards.
Experience training MoE architectures.
A history of mentoring engineers and serving as a go-to technical expert, rather than functioning primarily as a people manager or strategy owner.

Role Details

Location: San Francisco, CA (onsite)

Compensation: USD 180,000 - 298,000 per yearly

Minimum Experience: 1 years

View full job page →

6d ago

Lead Machine Learning Engineer

AeroVironment Centreville, VA

Centreville, VA

6d ago

$158k - $241k/yr Full time Manager Artificial Intelligence Automation

Apply Now

Job Description

AeroVironment’s Learning and Active Perception (LEAP) group builds computer vision and related analytics for defense and commercial applications. In this Lead Machine Learning Engineer role (onsite in Centreville, VA), you will work as a strong individual contributor and technical leader, guiding end-to-end development from requirements through test, evaluation, validation, and verification. You will also mentor more junior engineers while partnering across teams to help deliver practical ML capabilities.

What you’ll do

Lead the design and development of computer vision and machine learning algorithms for challenging problems such as detection, classifying, localizing, and tracking objects of interest across stationary and mobile sensor platforms.
Stay current with relevant academic literature, commercial products, and open source projects to keep solutions aligned with the latest capabilities.
Advise on and support software infrastructure and DevOps processes that enable ML development and deployment needs.
Implement ML algorithms and validate designs through experiments designed for the specific problem areas.
Evaluate trained models by analyzing algorithm behavior and errors, then design strategies to address failure modes.
Select appropriate datasets and data representations, and supervise data acquisition when required.
Manage available resources including hardware, data, and personnel to meet deadlines, coordinating with others and between teams as needed.
Perform other duties as assigned.

Qualifications

Bachelor’s degree in computer science/engineering, Mathematics, Physics, or related Science or Engineering discipline with 5+ years of relevant experience in machine learning and/or computer vision.
Minimum 8-12 years of relevant experience.
Advanced degree preferred.
Familiarity with C/C++ or Python (required).
Demonstrated ability to design, implement, and troubleshoot complex systems.
Proficiency with a deep learning framework, preferably PyTorch.
Proficiency with scikit-learn and pandas.
Proficiency with OpenCV.
Proficiency with Linux.
Familiarity with ML standards such as ONNX.
Ability to select hardware for specific ML applications.
Expertise in visualizing and manipulating big datasets.
Experience working with DoW and Intelligence Community (IC) organizations on advanced technology, research, or AI/ML initiatives.
Familiarity with classified spaces.
Familiarity working in office software and computer-based productivity tools.
Ability to work within defined requirements with minimal supervision.
Team player with leadership qualities, strong verbal and written communication, and effective problem-solving and analytical skills.
Ability to excel in a fast-paced, deadline-driven environment with small teams covering a broad variety of duties.
Demonstrated ownership of work products and a commitment to learning from mistakes to improve personal and team performance.
Must be able to work in an office environment (constant) and sit/stand for long periods, and talk, hear, and use hands and fingers to operate a computer keyboard (frequent).
U.S. Citizenship required.
Ability to obtain at minimum a Secret level DoD security clearance.
Willingness to travel frequently, up to 40%.

Core technologies

C/C++, Python
PyTorch
scikit-learn, pandas
OpenCV
Linux
ONNX

Benefits

Medical, dental, vision
401K with company matching
9/80 work schedule
Paid holiday shutdown

Compensation and location

Location: Centreville, VA (onsite)

Salary range: USD 158,000 - 241,000 per year

ITAR requirement: This position requires access to information subject to compliance with the International Traffic Arms Regulations (ITAR) and/or the Export Administration Regulations (EAR).

Clearance level: No Clearance

View full job page →

6d ago

Lead Machine Learning Engineer

AeroVironment Centreville, VA

Centreville, VA

6d ago

$158k - $241k/yr Full time Manager Ai Ml Artificial Intelligence

Apply Now

Job Description

AeroVironment’s MacCready Works division is seeking a Lead Machine Learning Engineer to support the LEAP group, focusing on computer vision and machine learning algorithm development for both defense and commercial applications. In this onsite role in Centreville, VA, you will own technical work across the machine learning lifecycle, from requirements through validation and verification testing, while mentoring junior engineers.

What you’ll do

Lead the design and development of computer vision and ML algorithms to tackle technically challenging problems such as detection, classification, localization, and tracking of objects of interest from stationary and mobile sensor platforms.
Stay current with academic literature, commercial products, and open source projects to keep solutions aligned with leading-edge approaches.
Advise on and support the design of software infrastructure and DevOps processes to support ML needs.
Implement ML algorithms and validate design choices through a set of purpose-designed experiments.
Evaluate model performance and analyze errors, then develop strategies to overcome observed limitations.
Select datasets and data representations, including supervising data acquisition when required.
Manage available resources, including hardware, data, and personnel, to meet deadlines; coordinate activities with a Team Lead and others, as additional duties require.

What you bring

Bachelor’s degree in computer science/engineering, Mathematics, Physics, or a related Science or Engineering discipline with 5+ years of relevant experience in machine learning and/or computer vision.
Minimum 10 to 12 years of relevant experience.
Advanced degree preferred.
Familiarity with C/C++ or Python required.
Demonstrated ability to design, implement, and troubleshoot complex systems.
Proficiency with a deep learning framework, preferably PyTorch.
Proficiency with scikit-learn and pandas.
Proficiency with OpenCV and Linux.
Familiarity with ML standards such as ONNX.
Ability to select hardware for specific ML applications.
Expertise in visualizing and manipulating big datasets.
Experience working with DoW and Intelligence Community (IC) organizations on advanced technology, research, or AI/ML initiatives.
Familiarity working in classified spaces.
Familiarity with office software and computer-based productivity tools.
Ability to work within defined requirements under minimal supervision and support team success with leadership qualities.
Strong verbal and written communication, along with problem-solving, analytical, and interpersonal skills.
Ability to excel in a fast-paced, deadline-driven environment with small teams and broad responsibilities.
Ability to work in an office environment (Constant), and to sit and stand for long periods and use hands/fingers to operate a computer keyboard (Frequent).
U.S. Citizenship required.
Active Secret level DoD security clearance with the ability to obtain a TS/SCI clearance.
Willingness and ability to travel occasionally (up to 20%).

Technologies

C/C++, Python, PyTorch
scikit-learn, pandas
OpenCV, Linux, ONNX
DevOps

Compensation and schedule

Salary: USD 158,000 - 241,000 per year
Location: Centreville, VA (onsite)
Work schedule: a 9/80 work schedule

Benefits

Medical
Dental vision
401K with company matching
Paid holiday shutdown

ITAR requirement: This position requires access to information subject to ITAR and/or EAR. Applicants must qualify as a U.S. person under ITAR and the EAR, or be a person approved for an export license by the relevant governing agency. U.S. person includes U.S. citizens, U.S. lawful permanent residents (green card holders), or protected individuals such as refugees or asylees. U.S. Citizenship is required.