Remote JAX Jobs

Remote JAX jobs hiring now at vetted startups and product companies on Alion, updated daily. Average salary around $205,554/year. Work from anywhere: fully remote positions with clear salary, time-zone and visa details. Browse and apply today.

AI/ML
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Executive
Networking
Product
Design
Analytics
Support
Enterprise Apps
Quantum

Jumio
jumio.com
Jumio is an IT & Services company that provides service related to online protection for businesses, specializing in online ID verification, card scanning, mobile payments, computer vision, credentials management, and KYC.
jumio.com • HQ: Sunnyvale, United States • Cybersecurity • Fraud Detection • Identity Management • Est. 2010
HQ: Sunnyvale, United States • Cybersecurity • Fraud Detection • Identity Management • Est. 2010
Verified live · 1 day ago 2 months ago

Machine Learning Engineer - (Computer Vision)
≈ $102k – $221k per year (Estimated) • Senior • 5+ years exp • Montreal • Remote (Canada) • Internship • Senior
Python
Databases
FAISS
Milvus
AI/ML
Computer Vision
JAX
Knowledge Distillation
ONNX
OpenCV
Pillow
PyTorch
Quantization
Synthetic Data
TensorFlow
TensorRT
Amazon SageMaker
Diffusion Models
TFLite
Mobile
Core ML
DevOps
Amazon EC2
Amazon EKS
AWS
Kubernetes
Apply
Verified live · 1 day ago 2 months ago

Machine Learning Engineer - IV (Computer Vision)
≈ $29k – $72k per year (Estimated) • Senior • 5+ years exp • Remote (India) • Internship • Senior
Python
Databases
FAISS
Milvus
AI/ML
Computer Vision
JAX
Knowledge Distillation
ONNX
OpenCV
Pillow
PyTorch
Quantization
Synthetic Data
TensorFlow
TensorRT
Amazon SageMaker
Diffusion Models
TFLite
Mobile
Core ML
DevOps
Amazon EC2
Amazon EKS
AWS
Kubernetes
Apply
Verified live · 1 day ago 2 months ago

Machine Learning Engineer
≈ $55k – $138k per year (Estimated) • Senior • 5+ years exp • Remote (Austria) • Internship • Senior
Python
Databases
FAISS
Milvus
AI/ML
Computer Vision
JAX
Knowledge Distillation
ONNX
OpenCV
Pillow
PyTorch
Quantization
Synthetic Data
TensorFlow
TensorRT
Amazon SageMaker
Diffusion Models
TFLite
Mobile
Core ML
DevOps
Amazon EC2
Amazon EKS
AWS
Kubernetes
Apply
Report

Gateworth Group
gateworth.com
Gateworth Group is an executive search and talent acquisition firm based in Dubai, specializing in recruitment across technology, finance, engineering, and corporate operations. The company connects multinational corporations, single-family offices, and emerging tech enterprises with specialized professionals and leadership talent. Through tailored recruitment services and market intelligence, it helps organizations source critical skill sets to drive regional and global business growth.
gateworth.com • Dubai • Professional Services
Dubai • Professional Services
2 months ago

Senior AI Engineer
Senior • Dubai • Remote (United Arab Emirates) • Relocation • Master's Degree • Full-Time • Senior
Fine-tuning
JAX
LLM
Multimodal AI
PyTorch
RAG
TensorFlow
Transformers
Hugging Face
AI Agents
Apply
Report

Deeplab
deeplab.ai
Empower your business with deep intelligenceto make better decisions.. We design and develop scalable ML infrastructure solutions employing cutting-edge technology.
deeplab.ai • Athens • Artificial Intelligence • Machine Learning
Athens • Artificial Intelligence • Machine Learning
Verified live · 2 days ago 2 months ago

R&D Machine Learning Engineer
Middle • 4+ years exp • Athens • Remote (Greece) • Middle • English
Python
AI/ML
Computer Vision
LLM
PEFT
LLM Evaluation
AI Agents
RAG
Fine-tuning
JAX
LoRA
MLFlow
NLP
PyTorch
TensorFlow
Time Series Forecasting
Weights & Biases
DevOps
CI/CD
Docker
Git
HPC
Apply
Report

Softtek
softtek.com
Softtek is a Mexican technology company founded in 1982 and headquartered in Monterrey. It provides software development, digital transformation and managed services worldwide. The company pioneered the nearshore delivery model from Latin America.
softtek.com • HQ: Monterrey, Mexico • Information Technology • 1001-5000 employees • Est. 1982
HQ: Monterrey, Mexico • Information Technology • 1001-5000 employees • Est. 1982
Verified live · 2 hours ago 2 months ago

Murex MRB senior developer
≈ $90k – $221k per year (Estimated) • Senior • 5+ years exp • Remote (Spain) • Bachelor's Degree • Senior • Spanish: C1
Java
Java
Apache Camel
Apache Tomcat
Maven
Quarkus
Spring Boot
Spring Framework
Databases
Apache Kafka
MySQL
AI/ML
JAX
DevOps
Docker
Git
Jenkins
Kubernetes
OpenShift
Cybersecurity
SonarQube
QA
Insomnia
Apply
Report

Valsoft Corporation
valsoftcorp.com
Valsoft Corporation is a Canadian software holding company specializing in the acquisition, development, and management of vertical market software (VMS) businesses. Headquartered in Montreal, Quebec, the company operates under a decentralized, "hold-forever" philosophy, partnering with niche software firms to drive long-term growth and stability.
valsoftcorp.com • HQ: Montreal, Canada • Information Technology • Software • 1001-5000 employees
HQ: Montreal, Canada • Information Technology • Software • 1001-5000 employees
Verified live · 12 hours ago 3 months ago

Senior Java Developer - Chordline Health
≈ $136k – $225k per year (Estimated) • Senior • Remote (United States) • Full-Time • Senior
JavaScript
SQL
Java
Node JS
Java
Apache Tomcat
Jackson
Node JS
Dotenv
Databases
MS SQL
AI/ML
JAX
Phi
Frontend
JQuery
Mobile
MVVM
DevOps
CI/CD
AWS
Azure
Docker
GCP
GitHub Actions
TeamCity
GitHub
Cybersecurity
HIPAA
QA
Playwright
Swagger
Apply
Report

MacPaw
macpaw.com
MacPaw is a software company headquartered in Kyiv, Ukraine, and founded in 2008 by Oleksandr Kosovan. The company develops consumer applications for macOS and iOS including CleanMyMac, the Setapp subscription app service, Encrypto, and ClearVPN, sold directly to users worldwide. It employs several hundred people, has continued operating through the war in Ukraine, and runs the MacPaw Foundation alongside its commercial work.
macpaw.com • HQ: Kyiv, Ukraine • Operating Systems • Endpoint Security • Laptops • Consumer Electronics • Productivity Software • Software • Cybersecurity • Est. 2008
HQ: Kyiv, Ukraine • Operating Systems • Endpoint Security • Laptops • Consumer Electronics • Productivity Software • Software • Cybersecurity • Est. 2008
Verified live · 1 day ago 3 months ago

AI Research Scientist
Remote (Ukraine) • Master's Degree
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
Fine-tuning
JAX
Knowledge Distillation
LLM
LoRA
MLX ML
NLP
PyTorch
Quantization
TensorFlow
PEFT
Edge AI
Mixture of Experts
Management
Jira
Apply
Report

Workato
workato.com
Workato provides an enterprise automation platform combining integration and workflow orchestration. Business and technical teams build recipes connecting hundreds of applications. Governance controls let central teams manage automation built across departments.
workato.com • HQ: Mountain View, United States • Software • Artificial Intelligence • Middleware Software • 1001-5000 employees • Est. 2013
HQ: Mountain View, United States • Software • Artificial Intelligence • Middleware Software • 1001-5000 employees • Est. 2013
Verified live · 3 min ago 3 months ago

Senior AI Research Scientist
≈ $148k – $266k per year (Estimated) • Senior • San Francisco • Remote (United States) • Master's Degree • Senior
CUDA
CUDA Toolkit
JAX
LLM
PyTorch
RAG
Reinforcement Learning
AI Agents
Model Context Protocol
Apply
Report

Twilio
twilio.com
Twilio is a leading cloud communications platform that enables software developers to programmatically build, scale, and integrate real-time communication capabilities into applications via application programming interfaces (APIs). Headquartered in San Francisco, California, the company provides infrastructure for automated SMS messaging, voice calls, WhatsApp integration, video, and two-factor authentication without requiring traditional telecom hardware.
twilio.com • HQ: San Francisco, United States • Telecommunications • Software • Online Communication • 1001-5000 employees • Est. 2008
HQ: San Francisco, United States • Telecommunications • Software • Online Communication • 1001-5000 employees • Est. 2008
Verified live · 3 hours ago 3 months ago

Machine Learning Engineer
≈ $52k – $155k per year (Estimated) • Junior • 2+ years exp • Remote (Ireland) • Bachelor's Degree • Junior
Python
AI/ML
AutoGen
CrewAI
Fine-tuning
JAX
LangGraph
LLM
NLP
NLTK
Prompt Engineering
PyTorch
TensorFlow
Transformers
LangChain
Hugging Face
LLMOps
AI Agents
Mobile
Twilio
DevOps
AWS
Azure
GCP
Apply
Verified live · 3 hours ago 3 months ago

Machine Learning Engineer
≈ $34k – $100k per year (Estimated) • Junior • 2+ years exp • Remote (Spain) • Bachelor's Degree • Junior
Python
AI/ML
AutoGen
CrewAI
Fine-tuning
JAX
LangGraph
LLM
NLP
NLTK
Prompt Engineering
PyTorch
TensorFlow
Transformers
LangChain
Hugging Face
LLMOps
AI Agents
Mobile
Twilio
DevOps
AWS
Azure
GCP
Apply
Report

Hugging Face
huggingface.co
Hugging Face is an American company, founded in 2016 and headquartered in New York, that runs the largest open platform for sharing machine learning models, datasets and demonstration applications. Its Hub hosts more than a million public models and is often described as the GitHub of machine learning, while its open-source libraries such as Transformers, Diffusers and Datasets are standard tools in research and production pipelines. The company earns revenue from paid Hub features, managed inference endpoints and enterprise deployments, and works with cloud and chip vendors to make open models easy to run anywhere.
huggingface.co • HQ: New York, United States • Software • MLOps • Machine Learning • LLM & Generative AI • Artificial Intelligence • 201-500 employees • Est. 2016
HQ: New York, United States • Software • MLOps • Machine Learning • LLM & Generative AI • Artificial Intelligence • 201-500 employees • Est. 2016
Open 95 daysVerified live · 2 days ago 4 months ago

Open-Source Machine Learning Engineer - US Remote
≈ $140k – $307k per year (Estimated) • New York • Remote (United States) • Full-Time
Python
AI/ML
Accelerate
Gradio
JAX
PyTorch
TensorFlow
Transformers
vLLM
Hugging Face
Fine-tuning
DevOps
GitHub
Management
Slack
Apply
Open 95 daysVerified live · 2 days ago 4 months ago

Open-Source Machine Learning Engineer - EMEA Remote
≈ $69k – $165k per year (Estimated) • Paris • Remote (EMEA) • Full-Time
Python
AI/ML
Accelerate
Gradio
JAX
PyTorch
TensorFlow
Transformers
vLLM
Hugging Face
Fine-tuning
DevOps
GitHub
Management
Slack
Apply
Report

Instacart AI
instacart.ai
Instacart's enterprise AI platform provides specialized artificial intelligence solutions designed to modernize online and in-store grocery retail. The technology suite includes white-label conversational shopping assistants, computer-vision inventory tracking, and automated catalog enrichment tools trained on extensive historical retail data. By integrating smart hardware and predictive analytics into existing supermarket networks, it enables grocers to streamline operations, enhance inventory visibility, and deliver personalized shopping experiences.
instacart.ai • San Francisco • Chicago • Boston • Wayne • Allentown
San Francisco • Chicago • Boston • Wayne • Allentown
Open 95 daysVerified live · 1 hour ago 4 months ago

Senior Machine Learning Engineer II, Ads Response Prediction
≈ $151k – $271k per year (Estimated) • Equity • Senior • Remote (United States) • Master's Degree • Full-Time • Senior
Python
SQL
AI/ML
LLM
Transformers
PEFT
Embeddings
Fine-tuning
JAX
LoRA
Pandas
PyTorch
Ray
Spark
TensorFlow
Transfer Learning
Mixture of Experts
Apply
Report

SandboxAQ
sandboxaq.com
SandboxAQ leverages the compound effects of AI and advanced computing to address some of the biggest challenges impacting society. SandboxAQ technologies include AI simulation, cryptography management for cybersecurity, and AI sensing for global organizations.
sandboxaq.com • HQ: Palo Alto, United States • Cybersecurity • Quantum Science & Computing • Artificial Intelligence • Est. 2016
HQ: Palo Alto, United States • Cybersecurity • Quantum Science & Computing • Artificial Intelligence • Est. 2016
Verified live · 1 min ago 4 months ago

Senior Machine Learning Engineer, AI Generation Engine
$86k – $151k per year • Senior • 5+ years exp • Remote (Canada) • Bachelor's Degree • Full-Time • Senior
Python
AI/ML
Braintrust
Fine-tuning
JAX
MLFlow
NumPy
Pandas
PyTorch
TensorFlow
Weights & Biases
Edge AI
Pre-training
AI Agents
DevOps
CI/CD
AWS
GCP
Apply
Verified live · 1 min ago 4 months ago

Staff Machine Learning Engineer, AI Generation Engine
$105k – $186k per year • Staff+ • 8+ years exp • Remote (Canada) • Bachelor's Degree • Full-Time • Staff
Python
AI/ML
Braintrust
Fine-tuning
JAX
MLFlow
NumPy
Pandas
PyTorch
TensorFlow
Weights & Biases
Edge AI
Pre-training
AI Agents
DevOps
CI/CD
AWS
GCP
Apply
Top 25% payVerified live · 1 min ago 4 months ago

Senior Machine Learning Engineer, AI Generation Engine
$131k – $246k per year • Senior • 5+ years exp • Remote (United States) • Bachelor's Degree • Full-Time • Senior
Python
AI/ML
Braintrust
Fine-tuning
JAX
MLFlow
NumPy
Pandas
PyTorch
TensorFlow
Weights & Biases
Edge AI
Pre-training
AI Agents
DevOps
CI/CD
AWS
GCP
Apply
Report

Binance
binance.com
Binance is the world's largest cryptocurrency exchange by trading volume, founded in 2017 by Changpeng Zhao and Yi He. The platform offers spot, margin and derivatives trading across hundreds of digital assets, alongside staking, savings products, payments, an institutional custody arm and a self-custodial Web3 wallet. The group also created BNB Chain, one of the most used smart contract networks, and now operates under a licensed regional structure after a 2023 settlement with United States authorities that installed new leadership and compliance oversight.
binance.com • HQ: Abu Dhabi, United Arab Emirates • Mining & Staking • Crypto Payments • Web3 & dApps • Blockchain & Crypto • Cryptocurrencies • Crypto Exchanges • 1001-5000 employees • Est. 2017
HQ: Abu Dhabi, United Arab Emirates • Mining & Staking • Crypto Payments • Web3 & dApps • Blockchain & Crypto • Cryptocurrencies • Crypto Exchanges • 1001-5000 employees • Est. 2017
Verified live · 11 hours ago 4 months ago

Quantitative Researcher (Fresh STEM PhD graduates are welcome)
Hong Kong • Remote (Taiwan, Hong Kong) • PhD
Python
AI/ML
AI Agents
JAX
LLM
NumPy
PyTorch
Time Series Forecasting
Apply
Report

Zencore
zencore.dev
Zencore is a leading cloud consulting and services firm. Above all else we value experience, hard work and the ability to keep things in perspective and have fun as we ensure our customers are delighted with the work we do.
zencore.dev • London • Mexico City • New York • Copenhagen • Berlin • Blockchain & Crypto • Data & Analytics • Information Technology • Est. 2013
London • Mexico City • New York • Copenhagen • Berlin • Blockchain & Crypto • Data & Analytics • Information Technology • Est. 2013
Open 137 days 5 months ago

Principal Architect, AI/ML
≈ $139k – $249k per year (Estimated) • Lead • Remote (United States) • Master's Degree • Full-Time • Architect
AI Agents
Claude
Fine-tuning
Gemini
Google ADK
JAX
LangChain
Langfuse
LangGraph
LangSmith
Llama
LLM
LoRA
Mistral
PyTorch
Quantization
TensorRT
TensorRT-LLM
Vertex AI
vLLM
PEFT
LLM Guardrails
TPU
DevOps
AWS
Azure
GCP
Google GKE
Kubernetes
Cybersecurity
GDPR
Apply
Open 202 days 7 months ago

Principal Architect, AI/ML
≈ $80k – $190k per year (Estimated) • Lead • Remote (United Kingdom) • Master's Degree • Full-Time • Architect
AI Agents
Claude
Fine-tuning
Gemini
Google ADK
JAX
LangChain
Langfuse
LangGraph
LangSmith
Llama
LLM
LoRA
Mistral
PyTorch
Quantization
TensorRT
TensorRT-LLM
Vertex AI
vLLM
PEFT
LLM Guardrails
TPU
DevOps
AWS
Azure
GCP
Google GKE
Kubernetes
Cybersecurity
GDPR
Apply
Report

D-Wave
dwavequantum.com
D-Wave is an industry leader in the development and production of quantum computing systems, software, and services. As the first commercial supplier of quantum computers, D-Wave is at the forefront of the dual development of quantum annealing and gate-model systems. D-Wave helps companies use quantum tech to solve real-world problems right now.
dwavequantum.com • HQ: Palo Alto, United States • Hardware • Artificial Intelligence • Science & Engineering • Quantum Computing Hardware • Quantum Science & Computing
HQ: Palo Alto, United States • Hardware • Artificial Intelligence • Science & Engineering • Quantum Computing Hardware • Quantum Science & Computing
Verified live · 12 hours ago 6 months ago

Staff Machine Learning Research Developer [430]
$105k – $158k per year • Staff+ • 6+ years exp • Remote (United States, Canada) • PhD • Full-Time • Staff
C++
C++
TensorFlow C++
AI/ML
JAX
Kubeflow
TensorFlow
Vertex AI
Apply
Report

Machinify
machinify.com
Machinify is an artificial intelligence software company headquartered in Palo Alto, California, and founded in 2016. The company provides an AI-powered operating system designed to automate healthcare payment integrity processes, including claims auditing, subrogation, and pharmacy payment accuracy. It serves major health plans across the United States, managing data for over 270 million members to identify cost savings and reduce administrative friction for payers and providers.
machinify.com • HQ: Palo Alto, United States • Business Process Automation (BPA) • Software • Insurance • Financial Services • Machine Learning • Health Care • Digital Health • Artificial Intelligence • Est. 2016
HQ: Palo Alto, United States • Business Process Automation (BPA) • Software • Insurance • Financial Services • Machine Learning • Health Care • Digital Health • Artificial Intelligence • Est. 2016
Top 25% payVerified live · 2 min ago 7 months ago

Staff Data Scientist | NLP
$180k – $230k per year • Staff+ • 3+ years exp • Remote (United States) • Staff
Python
SQL
Databases
pgvector
Pinecone
Weaviate
PostgreSQL
AI/ML
Fine-tuning
JAX
LLM
LoRA
NLP
PyTorch
QLoRA
RAG
Semantic Search
Sentence-Transformers
spaCy
Transformers
PEFT
AI Agents
Hugging Face
Human-in-the-Loop
LLM Evaluation
Semantic Search
Multimodal AI
DevOps
AWS
Azure
GCP
Vector
Analytics
A/B Testing
Apply
Report

Runway
runway.team
Runway is a mobile release management company headquartered in San Francisco, California, and founded in 2020. The company builds a platform that coordinates app store releases, automating build tracking, regression checks, staged rollouts, and the cross-team communication that surrounds a mobile launch. It integrates with the app stores, continuous integration systems, issue trackers, and chat tools that mobile teams already use, and went through Y Combinator.
runway.team • HQ: San Francisco, United States • Business Process Automation (BPA) • Project Management • Information Technology • DevOps • Software • Mobile App Development • Est. 2020
HQ: San Francisco, United States • Business Process Automation (BPA) • Project Management • Information Technology • DevOps • Software • Mobile App Development • Est. 2020
Open 234 daysTop 25% pay 8 months ago

Member of Technical Staff, Research Engineer
$270k – $370k per year • Staff+ • 4+ years exp • Remote (United States) • Full-Time • Staff
JAX
Multimodal AI
PyTorch
Runway
Post-training
SFT
Apply
Open 691 dayTop 25% pay 2 years ago

Research Engineer, Data Foundations
$270k – $370k per year • Middle • 4+ years exp • Remote (United States) • Full-Time • Middle
Multimodal AI
Synthetic Data
JAX
PyTorch
Ray
Runway
DevOps
Kubernetes
Apply
Report

Avride
avride.ai
Avride develops self-driving cars and sidewalk delivery robots, and was created in 2024 when Nebius spun out the former Yandex autonomous vehicle division. Its delivery robots operate on American university campuses and with food delivery platforms, while its robotaxi programme runs with Hyundai and Uber. Engineering is based in Austin, Texas.
avride.ai • HQ: Austin, United States • Transportation & Logistics • Smart Mobility • Autonomous Driving • 201-500 employees • Est. 2024
HQ: Austin, United States • Transportation & Logistics • Smart Mobility • Autonomous Driving • 201-500 employees • Est. 2024
Open 376 daysVerified live · 2 hours ago 1 year ago

Machine Learning Engineer – Motion Planning & Prediction
≈ $138k – $301k per year (Estimated) • Austin • Remote (United States) • Relocation
C++
Python
Rust
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
NLP
PyTorch
TensorFlow
AI Agents
Robotics
Motion Planning
SLAM
Apply
Report

AE Studio
ae.studio
AE Studio is a bootstrapped, ~150-person applied AI studio founded in 2016. We make high-stakes AI reliable: evals, red-teaming, guardrails, and continuous observability, with the data foundation that fixes failures at the root.
ae.studio • Los Angeles • Artificial Intelligence • Est. 2016
Los Angeles • Artificial Intelligence • Est. 2016
Open 1967 daysVerified live · 1 day ago 5 years ago

Applied AI Data Scientist
≈ $136k – $297k per year (Estimated) • Remote (United States, Brazil)
Python
AI/ML
Computer Vision
Cursor
Fine-tuning
JAX
LLM
NLP
PyTorch
RAG
TensorFlow
Time Series Forecasting
Apply
Open 1967 daysVerified live · 1 day ago 5 years ago

Applied AI Data Scientist (Brazil)
Remote (Brazil)
Python
AI/ML
Computer Vision
Cursor
Fine-tuning
JAX
LLM
NLP
PyTorch
RAG
TensorFlow
Time Series Forecasting
Apply
Report
Remote JAX Jobs
Frequently asked questions
How many Remote JAX jobs are available on Alion?
There are currently 75 active listings from 39 hiring companies on Alion, refreshed daily.
What salary do Remote JAX jobs pay?
The average salary is around $205,554 per year based on the current listings.
Do Remote JAX jobs offer relocation or visa sponsorship?
Many listings also offer relocation packages or visa sponsorship - use the filters to narrow the list down to what you need.
How do I apply for Remote JAX jobs?
Open any listing on Alion and apply directly with the employer through the provided link - no lengthy sign-up required.