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
Most Relevant
Found 42 jobs
Most Relevant
AI/ML Research Engineer, LLM Post-Training & Evaluation
Innodatat Services Inc.
Information Technology
Data Services
Publishing
RemoteCAD$81k - CAD$178k
Full-time
Mid-level
AI Engineer
H1B Visa Sponsor
As an AI/ML Research Engineer, you will optimize technical foundations for model improvement and collaborate with various stakeholders. The role involves building fine-tuning workflows and integrating evaluation systems for large language models.
CAD$81k - CAD$178k
Python
PyTorch
JAX
+2 more
Senior Staff Machine Learning Engineer
Tapestry Energy
Energy
Renewable Energy
Sustainability
Mountain View, California – Hybrid$262k - $361k
Full-time
Senior
Engineer
As a foundational architect of Tapestry’s multi-year machine learning strategy, you will bridge cutting-edge AI research and the development of production ML/AI systems. You will partner with various teams to translate complex grid data into actionable insights.
$262k - $361k
PyTorch
JAX
TensorFlow
Machine Learning Engineer - IV (Computer Vision)
Jumio
security-and-investigations
security-guards-and-patrol-services
security-systems-services
RemoteFull-time
Mid-level
AI Engineer
As a Machine Learning Engineer IV, you will lead the design and development of computer vision systems for biometrics, focusing on face recognition. You will also mentor ML engineers and optimize models for production deployment.
Python
PyTorch
Tensorflow
+3 more
Machine Learning Engineer
Jumio
security-and-investigations
security-guards-and-patrol-services
security-systems-services
Remote
Full-time
Mid-level
AI Engineer
We’re looking for a Machine Learning Engineer with deep expertise in computer vision and biometrics to lead the design and scaling of face recognition systems in production.
Python
PyTorch
Tensorflow
+3 moreG
Senior Machine Learning Engineer, Artificial Intelligence (AI)
Ginastechjobs
Remote$170k - $190k
Full-time
Senior
AI Engineer
In this role, you will build and ship Machine Learning systems while collaborating closely with research, product, and engineering teams. This position is fully remote and focuses on delivering real user impact.
$170k - $190k
GPU-based training and inference systems
JAX
Python
+1 moreG
Principal Machine Learning Engineer, Artificial Intelligence (AI)
Ginastechjobs
San Francisco, California – Remote$170k - $200k
Full-time
Leadership
AI Engineer
The Principal Machine Learning Engineer will operate across training, inference, evaluation, and infrastructure, solving the hardest architectural and performance problems. This is a hands-on, high-impact role focused on depth.
$170k - $200k
PyTorch
JAX
DeepSpeed
+9 more
Senior Machine Learning Engineer
Freenome
Biotechnology
Healthcare
Genomics
Brisbane, California – Hybrid$162k - $227k
Hybrid
Senior
AI Engineer
H1B Visa Sponsor
Join Freenome as a Senior Machine Learning Research Engineer to design and build deep learning pipelines. Collaborate with interdisciplinary teams to optimize AI/ML systems using large-scale genomic data.
$162k - $227k
Python
Java
Julia
+21 moreT
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