Find Your Dream Remote Job

Search 100,000+ work from home jobs at top remote companies

Active filters:

Tech: JAX
Filters

Visa Sponsorship


Found 39 jobs

T
Research Intern
TogetherAid
Apply
philanthropy
United StatesOffice
$0k - $0k
Full-time
Entry-level
AI Research Scientist
As a Research Intern at Together AI, you will develop and implement novel machine learning methods, conduct experiments, and contribute to scientific publications. The role involves working on foundation models, training systems, and evaluation metrics in a collaborative environment.
$0k - $0k
PyTorch
JAX
BetterUp
Senior Machine Learning Engineer
BetterUp
Apply
Health and Wellness
Human Resources
Professional Training & Coaching
United StatesHybrid
$180k - $275k
Full-time
Senior
Engineer
H1B Visa Sponsor
UK Visa Sponsor
This role involves prototyping, fine-tuning, and deploying AI models, owning the production lifecycle, and working closely with product and science teams. The position is hybrid, with locations in several US cities, and offers a competitive salary and benefits.
$180k - $275k
Python
Jupyter
PyTorch
+2 more
Discord
Staff Software Engineer, Machine Learning (Safety)
Discord
Apply
Internet
Software
Social Media
United StatesRemote
$272k - $306k
Full-time
Leadership
Engineer
H1B Visa Sponsor
UK Visa Sponsor
This role involves designing and maintaining ML infrastructure, collaborating across teams to improve platform safety, and leading safety-focused ML projects. The engineer will serve as a technical lead, develop cutting-edge safety techniques, and influence company safety strategies.
$272k - $306k
Python
PyTorch
JAX
+1 more
Tahoebio AI
Senior Machine Learning Engineer
Tahoebio AI
Apply
Biotechnology
Artificial Intelligence
Healthcare
United StatesSouth San Francisco – Office
Full-time
Senior
Research Scientist
As a Senior Machine Learning Engineer, you will design and implement advanced ML models for drug discovery, working with cutting-edge biological data. You will collaborate with scientists to develop scalable, optimized neural network architectures.
PyTorch
TensorFlow
Keras
+6 more
FinsterAI
AI Engineer
FinsterAI
Apply
Artificial Intelligence
Software Development
Technology
United KingdomLondon – Remote
Full-time
Mid-level
AI Engineer
Finster AI is seeking an AI Engineer to build and research advanced AI solutions in finance, focusing on large language models and multi-modal AI agents. The role involves collaboration, research, and developing production-ready code in a fast-paced, innovative environment.
Python
PyTorch
JAX
+1 more
Waymo
Machine Learning Engineer, Perception Modeling
Waymo
Apply
Automotive
Technology
Artificial Intelligence
United StatesCalifornia – Remote
$170k - $216k
Full-time
Mid-level
Research Scientist
H1B Visa Sponsor
This role involves applying machine learning techniques to build perception systems for autonomous driving, working with large-scale data and models. The engineer will develop scalable training recipes, evaluate models, and collaborate with research teams to improve vehicle perception capabilities.
$170k - $216k
Python
PyTorch
JAX
M
Machine Learning Engineer
matchgroup
Apply
South KoreaSeoul – Office
Full-time
Mid-level
Analytics Engineer
As a Machine Learning Engineer at Match Group, you will research, develop, and deploy AI/ML models for various products, collaborating with cross-functional teams. The position focuses on applying state-of-the-art techniques to real-world problems in a dynamic environment.
PyTorch
TensorFlow
JAX
+1 more
Waymo
Staff TLM, Perception, Continuous Learning & Monitoring
Waymo
Apply
Automotive
Technology
Artificial Intelligence
United StatesRemote
$238k - $302k
Full-time
Leadership
Analytics Engineer
H1B Visa Sponsor
This role involves managing ML models, leading data selection and model updates, and monitoring deployed systems for Waymo's autonomous vehicle technology. The position requires experience in machine learning, computer vision, and team leadership.
$238k - $302k
Python
PyTorch
JAX
Waymo
Senior Software Engineer
Waymo
Apply
Automotive
Technology
Artificial Intelligence
United StatesCalifornia – Hybrid
$204k - $259k
Full-time
Senior
Engineer
H1B Visa Sponsor
This role involves collaborating with cross-functional teams to deploy ML models, develop CI/CD pipelines, and implement best practices in ML Ops. The position offers a salary range of $204,000 to $259,000 USD and is based in multiple US locations with a hybrid work model.
$204k - $259k
Python
C++
GCP
+13 more
OpenTeams
Senior Solutions Architect / Senior Solutions Engineer (Commercial)
OpenTeams
Apply
computer-software
artificial-intelligence
big-data
United StatesRemote
Full-time
Senior
Consultant
OpenTeams is hiring Senior Solution Architects and Engineers to lead AI solution design, client engagement, and community contributions. The role involves technical leadership, architecture development, and supporting open-source AI projects remotely.
PyTorch
JAX
Kubeflow
+13 more
Foundation Robotics
VLA Engineer
Foundation Robotics
Apply
Robotics
Construction Technology
Automation
United StatesSan Francisco – Remote
$80k - $1000k
Full-time
Mid-level
Engineer
Join Foundation as a VLA Engineer to work on advanced models for robotics, perception, and real-time control. Contribute to building scalable pipelines, deploying models, and collaborating across teams to advance general purpose robotics.
$80k - $1000k
PyTorch
JAX
GPU acceleration
+1 more
Ocean City Parasail
AI Performance Engineer
Ocean City Parasail
Apply
entertainment
business-services
nec
United StatesHybrid
Full-time
Mid-level
Engineer
This role involves developing and optimizing AI performance on distributed systems, including GPU kernels and Kubernetes deployments. It requires expertise in GPU computing, performance analysis, and open-source contributions to enhance AI workloads at enterprise scale.
CUDA
ROCm
XLA
+5 more
F
Associate Director / Director, AI & Machine Learning
Flagshippioneeringinc
Apply
United StatesOffice
$212k - $319k
Full-time
Senior
AI Engineer
This role involves leading the design and implementation of AI/ML systems, including generative models and causal inference, to accelerate drug discovery. The candidate will provide technical leadership, collaborate with cross-functional teams, and stay current with emerging AI advances.
$212k - $319k
PyTorch
TensorFlow
JAX
+1 more
Pure Storage
AI/HPC Pre-Sales Systems Engineer
Pure Storage
Apply
Information Technology and Services
Computer Hardware
Data Storage
United StatesOffice
$152k - $244k
Full-time
Mid-level
AI Engineer
H1B Visa Sponsor
UK Visa Sponsor
This role involves technical leadership in AI and HPC pre-sales, developing customer solutions, and collaborating with sales teams. The candidate will also create technical content, support RFP responses, and stay current with industry trends.
$152k - $244k
pytorch
tensorflow
jax
+14 more
Pulse Marketing Agency
Machine Learning Engineer
Pulse Marketing Agency
Apply
marketing-and-advertising
advertising
advertising-campaigns
United StatesSan Francisco, California – Office
$0k - $0k
Full-time
Mid-level
Engineer
Pulse is seeking a Machine Learning Engineer Intern to work on core ML challenges involving vision, NLP, and data infrastructure. The role involves training models, data curation, and collaborating with engineers in a fast-growing environment.
$0k - $0k
Python
PyTorch
JAX
+3 more
G
Machine Learning Engineer
goodfire
Apply
computer-software
United StatesSan Francisco, California – Remote
$200k - $400k
Full-time
Mid-level
Engineer
Join Goodfire as a Machine Learning Engineer to build tools and infrastructure for interpretable AI systems. Contribute to research, development, and deployment of cutting-edge AI interpretability technology.
$200k - $400k
Python
PyTorch
Jax
+1 more
D
Ad D Afc Bcc
d plus 2
Apply
book-publishing
United StatesSanta Clara – Office
$145k - $200k
Full-time
Mid-level
Engineer
Seeking a Machine Learning Engineer specialized in perception models for autonomous driving, focusing on sensor fusion and 3D scene understanding. The role involves designing, implementing, and optimizing perception algorithms using multimodal sensor data.
$145k - $200k
Python
PyTorch
TensorFlow
+7 more
OpenTeams
Senior Solutions Architect / Senior Solutions Engineer
OpenTeams
Apply
computer-software
artificial-intelligence
big-data
United StatesRemote
Full-time
Senior
Consultant
OpenTeams is hiring senior professionals to lead AI architecture and client engagement, focusing on open-source AI frameworks and cloud platforms. The role involves technical leadership, community participation, and supporting diverse organizations, including US federal agencies.
PyTorch
JAX
Kubeflow
+3 more
General Motors
Machine Learning Engineer
General Motors
Apply
Automotive
Manufacturing
Transportation
United StatesSunnyvale – Hybrid
$13k - $13k
Full-time
Mid-level
AI Engineer
H1B Visa Sponsor
UK Visa Sponsor
This is a full-time summer internship at GM in Sunnyvale, CA, where you will work on AI/ML projects related to autonomous vehicles. The role involves collaboration on perception models, deep learning pipelines, and applied research in robotics.
$13k - $13k
Python
PyTorch
TensorFlow
+4 more
OpenTeams
Senior Solutions Architect / Senior Solutions Engineer (Government)
OpenTeams
Apply
computer-software
artificial-intelligence
big-data
United StatesRemote
Full-time
Senior
Consultant
OpenTeams is hiring Senior Solution Architects and Engineers to lead AI solution design, client engagement, and community contributions, especially for US federal government projects. The role involves technical strategy, architecture development, and thought leadership in open-source AI.
PyTorch
JAX
Kubeflow
+3 more