Find Your Dream Remote Job
Search 100,000+ work from home jobs at top remote companies
All
Active filters:
Tech: JAX
Filters
Visa Sponsorship
JAX
Found 39 jobs
T
Research Intern
TogetherAid
philanthropy
Office$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
Senior Machine Learning Engineer
BetterUp
Health and Wellness
Human Resources
Professional Training & Coaching
Hybrid$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
Staff Software Engineer, Machine Learning (Safety)
Discord
Internet
Software
Social Media
Remote$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
Senior Machine Learning Engineer
Tahoebio AI
Biotechnology
Artificial Intelligence
Healthcare
South San Francisco – OfficeFull-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
AI Engineer
FinsterAI
Artificial Intelligence
Software Development
Technology
London – RemoteFull-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
Machine Learning Engineer, Perception Modeling
Waymo
Automotive
Technology
Artificial Intelligence
California – 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
Seoul – OfficeFull-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
Staff TLM, Perception, Continuous Learning & Monitoring
Waymo
Automotive
Technology
Artificial Intelligence
Remote$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
Senior Software Engineer
Waymo
Automotive
Technology
Artificial Intelligence
California – 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
Senior Solutions Architect / Senior Solutions Engineer (Commercial)
OpenTeams
computer-software
artificial-intelligence
big-data
RemoteFull-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
VLA Engineer
Foundation Robotics
Robotics
Construction Technology
Automation
San 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
AI Performance Engineer
Ocean City Parasail
entertainment
business-services
nec
HybridFull-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 moreF
Associate Director / Director, AI & Machine Learning
Flagshippioneeringinc
Office$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
AI/HPC Pre-Sales Systems Engineer
Pure Storage
Information Technology and Services
Computer Hardware
Data Storage
Office$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
Machine Learning Engineer
Pulse Marketing Agency
marketing-and-advertising
advertising
advertising-campaigns
San 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 moreG
Machine Learning Engineer
goodfire
computer-software
San 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 moreD
Ad D Afc Bcc
d plus 2
book-publishing
Santa 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
Senior Solutions Architect / Senior Solutions Engineer
OpenTeams
computer-software
artificial-intelligence
big-data
RemoteFull-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
Machine Learning Engineer
General Motors
Automotive
Manufacturing
Transportation
Sunnyvale – 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
Senior Solutions Architect / Senior Solutions Engineer (Government)
OpenTeams
computer-software
artificial-intelligence
big-data
RemoteFull-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