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.

You searched for “Remote JAX Jobs”, so this list shows: technical roles only · fully remote · the stack names JAX · known staffing agencies left out. Adjust the filters

AI/ML
Software
Data
Security
DevOps & Cloud
Leadership
Product
Design
Fintech
Hardware & Chips
Defense
Aerospace
Biotech
Energy
Robotics
Automotive
MedTech
Science

Netflix

netflix.com
netflix.com · HQ: Los Gatos, United States · Streaming & OTT Platforms · 1001-5000 employees · NASDAQ: NFLX · Trust A · Hiring 304, 24 new
HQ: Los Gatos, United States · Streaming & OTT Platforms · 1001-5000 employees · NASDAQ: NFLX · Trust A · Hiring 304, 24 new
Netflix is an entertainment service that streams television series, films and games across a wide variety of genres and languages. It reports over 300 million paid memberships in more than 190 countries, and it runs engineering teams for partner and client device technologies, with hiring in Warsaw. The company is headquartered in Los Gatos, California, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 11 hours ago Hiring now · 90%

Staff/ Principal Software Engineer (L6/L7): AI Platform Model Serving

≈ $235k – $451k per year (Estimated) · Equity · Lead · United States · Remote (United States) · Bachelor's Degree · Principal
vLLM
Triton Inference Server
SGLang
Speculative Decoding
Prompt Caching
LLM
DeepSpeed
Quantization
JAX
TensorRT
TensorRT-LLM
TGI
Transformers
CUDA
Hugging Face
NCCL
KV Cache
TensorFlow
PyTorch
Machine Learning
AI Agents
DevOps
GCP
Kubernetes
Azure
AWS
Analytics
A/B Testing
+18
Apply

Bright Vision Technologies

brightvisiontechnologies.com
brightvisiontechnologies.com · HQ: Bridgewater Township, United States · IT Consulting & Digital Transformation · IT Staffing & Staff Augmentation · 1001-5000 employees · Est. 2020 · Staffing · EOR · Hiring 504, 487 new
HQ: Bridgewater Township, United States · IT Consulting & Digital Transformation · IT Staffing & Staff Augmentation · 1001-5000 employees · Est. 2020 · Staffing · EOR · Hiring 504, 487 new
Bright Vision Technologies is an IT consulting, enterprise technology services, and workforce solutions enterprise. Headquartered in Bridgewater, New Jersey, United States, the minority-owned firm specializes in technology staffing, cybersecurity, application management, and digital product engineering. Founded in 2020, the enterprise delivers specialized staffing and IT services alongside proprietary automation and AI software - including its flagship enterprise talent intelligence platform, Lumina - serving clients across information technology, defense, healthcare, government, and manufacturing sectors.
No named decision-maker yet.
No careers mailbox known
Verified live · 7 hours ago 1 day ago Hiring now · 90%

Research Scientist – AI - Bright Vision Technologies

$100k – $150k per year · Senior · 6+ years exp · Remote (United States) · Master's Degree · Senior
Python
AI/ML
Fine-tuning
JAX
PyTorch
Machine Learning
Multimodal AI
RAG
AI Agents
Apply
Verified live · 7 hours ago 1 day ago Hiring now · 90%

GPU Applications Engineer - Bright Vision Technologies

$100k – $150k per year · Senior · 6+ years exp · Remote (United States) · Bachelor's Degree · Senior
C++
C++
LLVM
AI/ML
JAX
CUDA
Triton
NCCL
PyTorch
vLLM
TensorRT
MLIR
CUTLASS
DevOps
HPC
+4
Apply
Verified live · 7 hours ago 1 day ago Hiring now · 90%

AI Research Engineer - Bright Vision Technologies

$80k – $100k per year · Senior · 6+ years exp · Remote (United States) · Master's Degree · Senior
Python
AI/ML
Fine-tuning
JAX
PyTorch
Machine Learning
Multimodal AI
RAG
AI Agents
Apply

Mercor

mercor.com
mercor.com · HQ: San Francisco, United States · Recruiting & Staffing · Freelance & Talent Marketplaces · AI Training Data & Annotation · 1001-5000 employees · Est. 2023 · Trust A · Hiring 206, 19 new
HQ: San Francisco, United States · Recruiting & Staffing · Freelance & Talent Marketplaces · AI Training Data & Annotation · 1001-5000 employees · Est. 2023 · Trust A · Hiring 206, 19 new
Mercor is an American company founded in 2023 by three Thiel Fellows that recruits and pays domain experts to produce the training and evaluation data frontier artificial intelligence labs need. Its business shifted quickly from general talent matching to expert data work: as models became competent at ordinary tasks, the scarce input became judgement from practising doctors, lawyers, bankers and engineers, and Mercor sources, vets and pays those people at very large scale. Headquartered in San Francisco and valued in the billions within two years of founding, it pays out enormous sums annually to a contributor network rather than employing them.
No named decision-maker yet.
Careers mailbox is known
5 days ago Hiring now · 90%

MLOps Engineer - AI Systems

≈ $138k – $315k per year (Estimated) · Junior · 2+ years exp · San Francisco · Remote (United States) · Part-Time · Junior
vLLM
JAX
SGLang
TensorRT
TensorRT-LLM
CUDA
Triton
Megatron-LM
FSDP
TPU
AWS Trainium
MLIR
XLA
CUTLASS
KV Cache
PyTorch
LLM
Management
Stripe
+10
Apply
Top 25% pay 6 days ago Hiring now · 90%

AI Research Scientist - Fully Remote | Up to $200/hr

$300k – $400k per year · Remote (likely United States) · Contractor
Fine-tuning
Reinforcement Learning
JAX
PyTorch
Machine Learning
Apply
18 days ago Hiring now · 77%

AI Researcher - Fully Remote | Upto $120/hr

Remote (India) · Full-Time
JAX
Post-training
Pre-training
TensorFlow
Computer Vision
PyTorch
LLM
Machine Learning
Apply

Premera

premera.com
premera.com · Columbia · Spokane · Mountlake Terrace · 501-1000 employees · Est. 1945 · Trust B · Hiring 35, 5 new
Columbia · Spokane · Mountlake Terrace · 501-1000 employees · Est. 1945 · Trust B · Hiring 35, 5 new
Premera Blue Cross is an independent health insurance provider, serving businesses and residents of Washington State and Alaska as a licensee of the Blue Cross Blue Shield Association. The company offers a range of health plans including individual, family, employer-based, and Medicare Supplement plans. However, it has announced that it will exit the Medicare Advantage market as of January 1, 2025.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 20 min ago 5 days ago Hiring now · 90%

AI Engineer IV

$132k – $237k per year · Senior · 6+ years exp · United States · Remote (United States) · Bachelor's Degree · Full-Time · Senior
Python
AI/ML
Scikit-learn
JAX
Multimodal AI
Diffusion Models
MLX ML
Pandas
NumPy
Streamlit
Explainable AI
TensorFlow
Prompt Engineering
PyTorch
RAG
Analytics
Matplotlib
Management
Agile
+8
Apply

Mira Construction

miraconstruction.com
miraconstruction.com · HQ: Dubai, United Arab Emirates · Construction · Civil & Structural Engineering · 1001-5000 employees · Trust B · Hiring 124, 21 new
HQ: Dubai, United Arab Emirates · Construction · Civil & Structural Engineering · 1001-5000 employees · Trust B · Hiring 124, 21 new
Mira Construction is a general contracting and construction management firm that delivers residential, commercial, and institutional building projects. The company provides end-to-end project execution services, including pre-construction planning, site development, structural engineering, and interior finishing. By collaborating closely with architects and property owners, it focuses on delivering quality craftsmanship, cost-effective scheduling, and structural integrity tailored to client specifications.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 6 days ago Hiring now · 86%

Head of ML / Head of Quant Research (Crypto Trading)

≈ $72k – $161k per year (Estimated) · Head+ · London · Amsterdam · Remote (United Kingdom, Cyprus, Netherlands, Serbia, Georgia) · Full-Time · Architect · Russian (Optional) · English: A2 (Optional)
C++
Python
AI/ML
Machine Learning
Numba
JAX
NumPy
PyTorch
Apply
Verified live · 1 hour ago 3 months ago Fading · 33%

Senior ML Engineer / ML Lead

≈ $57k – $141k per year (Estimated) · Lead · Remote (Serbia, Cyprus, Poland, Spain, Georgia) · Full-Time · Staff
C++
Python
AI/ML
JAX
NumPy
PyTorch
Apply

Nebius

nebius.com
nebius.com · HQ: Amsterdam, Netherlands · Data Centers & Colocation · Cloud Platforms (IaaS & PaaS) · AI Compute & Inference · 1001-5000 employees · Trust B · Hiring 343, 18 new
HQ: Amsterdam, Netherlands · Data Centers & Colocation · Cloud Platforms (IaaS & PaaS) · AI Compute & Inference · 1001-5000 employees · Trust B · Hiring 343, 18 new
Nebius is an international technology company that specializes in building full-stack artificial intelligence infrastructure and high-performance cloud GPU platforms for AI model development. Headquartered in Amsterdam, Netherlands, the firm operates energy-efficient data centers across Europe and North America to provide scalable compute, storage, and software tools for machine learning workloads.
No named decision-maker yet.
Careers mailbox is known
6 days ago Likely open · 63%

Senior ML Engineer (Token Factory)

≈ $62k – $163k per year (Estimated) · Senior · Serbia · Remote (Serbia) · Senior
Python
AI/ML
DeepSeek
LoRA
Fine-tuning
Reinforcement Learning
JAX
Multimodal AI
Mixture of Experts
SFT
Speculative Decoding
LLM
Machine Learning
Flash Attention
Quantization
NLP
PEFT
FSDP
DevOps
CI/CD
+10
Apply
Verified live · 2 hours ago 5 months ago Long shot · 9%

ML Infrastructure Engineer

≈ $169k – $375k per year (Estimated) · Netherlands · United States · Remote (United States, Netherlands)
Python
AI/ML
vLLM
JAX
SGLang
TensorRT
CUDA
Megatron-LM
NCCL
ROCm
PyTorch
LLM
Machine Learning
DevOps
Docker
GCP
Kubernetes
Azure
AWS
+9
Apply

Reddit

reddit.com
reddit.com · HQ: San Francisco, United States · Social Networks · Online Forums & Community Platforms · 1001-5000 employees · Est. 2005 · NYSE: RDDT · Trust B · Hiring 154, 12 new
HQ: San Francisco, United States · Social Networks · Online Forums & Community Platforms · 1001-5000 employees · Est. 2005 · NYSE: RDDT · Trust B · Hiring 154, 12 new
Reddit is an American social platform founded in 2005 and organised into hundreds of thousands of user-created communities known as subreddits. Members post links, text, images and video, and the community ranks each submission through upvotes and downvotes, which decides what surfaces on the front page and inside each community. Headquartered in San Francisco and listed on the New York Stock Exchange since 2024, the company earns most of its revenue from advertising and has added data licensing deals that let AI developers train on its public conversation archive.
Steve Huffman · Co-Founder & Chief Executive Officer · decision-maker
Nellie Peshkov · Executive Officer · decision-maker
Harold Klaje · Executive Officer · decision-maker
Chris Slowe · Executive Officer · decision-maker
No careers mailbox known
Top 25% payVerified live · 5 hours ago Visa sponsor 7 days ago Likely open · 63%

Principal Data Scientist, User Ecosystem

$313k – $365k per year · Equity · Lead · United States · Remote (United States) · Principal
JAX
Knowledge Graph
Recommender Systems
PyTorch
Machine Learning
Apply

Zillow

zillow.com
zillow.com · HQ: Seattle, United States · Property Management · Mortgage Lending & Brokers · Real Estate Brokerage · PropTech · 1001-5000 employees · Est. 2006 · Trust A · Hiring 194, 11 new
HQ: Seattle, United States · Property Management · Mortgage Lending & Brokers · Real Estate Brokerage · PropTech · 1001-5000 employees · Est. 2006 · Trust A · Hiring 194, 11 new
Zillow Group is a Seattle-headquartered online real estate marketplace whose apps and websites let people search homes for sale and rent, estimate property values with the Zestimate and connect with agents, lenders and landlords. Founded in 2006 by former Microsoft and Expedia executives including Rich Barton and Lloyd Frink, the Nasdaq-listed company also runs Zillow Home Loans, Zillow Closing, rentals and advertising businesses, with engineering teams in Bengaluru and Mexico City. It recruits software and machine learning engineers, data scientists, product managers, mortgage loan officers, underwriting and closing staff, and sales and marketing specialists.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 1 day ago Visa sponsor 8 days ago Hiring now · 80%

Applied Research Scientist, World Model

$196k – $312k per year · Equity · United States · Remote (United States) · PhD · Full-Time
Python
AI/ML
JAX
World Models
PyTorch
Machine Learning
Reinforcement Learning
Post-training
LLM
AI Agents
+1
Apply
Open 126 daysVerified live · 1 day ago 4 months ago Long shot · 4%

Machine Learning Engineer

≈ $79k – $181k per year (Estimated) · Equity · Junior · 1+ year exp · Germany · Remote (Germany) · Full-Time · Junior
TypeScript
Python
AI/ML
JAX
arXiv
TensorFlow
Computer Vision
Spark
PyTorch
Machine Learning
DevOps
GCP
Kubernetes
CI/CD
AWS
+5
Apply

Parallel Partners

parallelpartners.com
parallelpartners.com · HQ: Charlotte, United States · Career Services · Executive Search · Recruiting & Staffing · Hiring 9
HQ: Charlotte, United States · Career Services · Executive Search · Recruiting & Staffing · Hiring 9
Parallel Partners is a technology recruitment firm headquartered in Charlotte, North Carolina. The agency places engineers, infrastructure specialists, and technology leaders into permanent roles at United States employers, with a growing concentration in artificial intelligence and mobile development positions. It works on both onsite searches in the Carolinas and fully remote placements across the country.
No named decision-maker yet.
Careers mailbox is known
$120k – $160k per year · San Francisco · Remote (United States) · Full-Time · Senior · English
Python
AI/ML
Embeddings
JAX
Reranking
Hybrid Search
Recommender Systems
PyTorch
Machine Learning
Apply
$120k – $160k per year · San Francisco · Remote (United States) · Full-Time · Senior · English
Python
AI/ML
Embeddings
JAX
Reranking
Hybrid Search
Recommender Systems
PyTorch
Machine Learning
Apply

KIEFER

kiefer.gr
kiefer.gr · HQ: Athens, Greece · AI Compute & Inference · Data Centers & Colocation · Foundation Models · Wind Energy · Hiring 10
HQ: Athens, Greece · AI Compute & Inference · Data Centers & Colocation · Foundation Models · Wind Energy · Hiring 10
KIEFER is an energy and advanced technology group headquartered in Athens, Greece. The company builds and operates renewable energy projects, principally wind, and runs a technology arm developing Sophea AI, a family of Greek language models, along with robotics and sovereign GPU cloud infrastructure. It is establishing Greece's first sovereign AI data centre, powered by renewable generation, in partnership with NVIDIA.
No named decision-maker yet.
Careers mailbox is known
Verified live · 13 min ago 8 days ago Hiring now · 90%

ML Engineer — ASR

Middle · 3+ years exp · Athens · Remote (Greece) · Full-Time · Middle · Greek · English
Python
AI/ML
Fine-tuning
JAX
Speech Recognition
TensorFlow
PyTorch
Voice Agents
Physical AI
Apply
Verified live · 13 min ago 8 days ago Hiring now · 90%

ML Engineer — Agents & LLMs

Middle · 3+ years exp · Athens · Remote (Greece) · Full-Time · Middle
Python
AI/ML
Reinforcement Learning
JAX
Function Calling
SFT
Post-training
TensorFlow
PyTorch
LLM
Physical AI
Reward Modeling
Robotics
Reinforcement Learning
+4
Apply

Clariti

claritisoftware.com
claritisoftware.com · Digital Government · Hiring 6, 1 new
Digital Government · Hiring 6, 1 new
Clariti's government software helps North America's largest and fastest growing communities deliver exceptional community development, permitting, and licensing experiences online. Our platform is built as an alternative to code-heavy, non-configurable systems that create technology barriers for governments to meet their community's evolving needs. We believe that organizations should be able to dictate how their software works.
No named decision-maker yet.
Careers mailbox is known
Verified live · 4 hours ago 8 days ago Hiring now · 83%

Senior Applied AI Engineer

≈ $90k – $173k per year (Estimated) · Senior · 7+ years exp · Canada · Remote (Canada) · Full-Time · Senior
Python
AI/ML
Fine-tuning
Multimodal AI
Diffusion Models
Model Distillation
JAX
Function Calling
LLM Guardrails
TensorFlow
Computer Vision
PyTorch
LLM
AI Agents
DevOps
Git
GCP
Azure
CI/CD
AWS
+10
Apply

Clariti

clariti.com
clariti.com · Sales & Marketing · Content Creation · Search Engine Optimization (SEO) · Hiring 5
Sales & Marketing · Content Creation · Search Engine Optimization (SEO) · Hiring 5
Clariti helps site owners organize and optimize their content for maximum growth.
No named decision-maker yet.
No careers mailbox known
11 days ago Hiring now · 75%

Senior Applied AI Engineer

≈ $109k – $251k per year (Estimated) · Senior · Remote (likely Canada) · Full-Time · Senior
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
Function Calling
Diffusion Models
NLP
LLM Guardrails
Interpretability
Model Distillation
TensorFlow
Computer Vision
PyTorch
LLM
Machine Learning
AI Agents
DevOps
Git
GCP
Azure
CI/CD
AWS
+13
Apply

Bjak

bjak.my
bjak.my · HQ: Petaling Jaya, Malaysia · InsurTech · Financial Comparison & Loan Marketplaces · 5000+ employees · Est. 2015 · Trust B · Hiring 1826, 23 new
HQ: Petaling Jaya, Malaysia · InsurTech · Financial Comparison & Loan Marketplaces · 5000+ employees · Est. 2015 · Trust B · Hiring 1826, 23 new
Bjak is a Malaysian insurance technology company founded in 2015 that operates the largest insurance comparison site in Southeast Asia, letting consumers compare and buy motor and travel cover online. It has extended beyond comparison into insurance distribution technology, claims processing and a broader financial services platform, and it develops much of its own infrastructure in house including document processing and pricing systems. Headquartered in Petaling Jaya near Kuala Lumpur, it operates across Malaysia, Thailand, Taiwan and other regional markets and remains privately held and profitable.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago 12 days ago Fading · 36%

Machine Learning Engineer, Applied AI

≈ $48k – $134k per year (Estimated) · Singapore · Remote (Singapore) · Full-Time
Python
AI/ML
Embeddings
Reranking
Hybrid Search
JAX
Recommender Systems
PyTorch
Machine Learning
Apply

Huntingcube

huntingcube.ai
huntingcube.ai · HQ: Gurgaon, India · Recruiting & Staffing · Recruiting Software & ATS · Hiring 46, 11 new
HQ: Gurgaon, India · Recruiting & Staffing · Recruiting Software & ATS · Hiring 46, 11 new
Hire top-tier tech talent across India with AI-powered precision. CubicAI delivers a 98% match rate, 4-day shortlisting, and verified profiles from a 1M+ talent pool. Post a job free.
No named decision-maker yet.
No careers mailbox known
12 days ago Hiring now · 75%

Machine Learning Engineer IV

≈ $23k – $53k per year (Estimated) · Senior · 5+ years exp · India · Remote (India) · Senior
Python
AI/ML
Airflow
OpenCV
Quantization
JAX
ONNX
TensorRT
Synthetic Data
Amazon SageMaker
Model Distillation
TensorFlow
Computer Vision
PyTorch
Machine Learning
DevOps
Amazon EKS
Amazon EC2
Docker
Kubernetes
AWS
+11
Apply

Lancedb

lancedb.com
lancedb.com · HQ: San Francisco, United States · Databases & DBMS · Data Warehouses & Lakehouses · Vector Search & RAG · 51-200 employees · Startup · Trust A · Hiring 12, 1 new
HQ: San Francisco, United States · Databases & DBMS · Data Warehouses & Lakehouses · Vector Search & RAG · 51-200 employees · Startup · Trust A · Hiring 12, 1 new
The multimodal lakehouse for AI, accelerating large-scale data curation and feature engineering so teams can build better models faster.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 4 hours ago 12 days ago Hiring now · 90%

Senior Software Engineer

$180k – $250k per year · Senior · 10+ years exp · Remote (AMER) · Full-Time · Senior
Rust
Scala
C++
Java
Python
Databases
ClickHouse
Apache Iceberg
Delta Lake
Presto
LanceDB
Apache Hudi
Trino
AI/ML
Hadoop
JAX
Multimodal AI
Midjourney
Flink
Runway
Ray
World Models
Spark
PyTorch
+14
Apply

YO AI Labs

yoailabs.com
yoailabs.com · HQ: San Francisco, United States · AI Evaluation & Observability · AI Training Data & Annotation · 5000+ employees · Trust A · Hiring 927, 39 new
HQ: San Francisco, United States · AI Evaluation & Observability · AI Training Data & Annotation · 5000+ employees · Trust A · Hiring 927, 39 new
YO AI Labs is an artificial intelligence platform specializing in AI model training, evaluation, and data annotation services. Headquartered in San Francisco, California, United States, the company connects domain experts and remote specialists across the globe to review, test, and benchmark generative AI model outputs. Through its decentralized workforce solutions and structured evaluation workflows, it enables frontier technology developers to enhance the accuracy, safety, and reliability of next-generation artificial intelligence systems.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 13 days ago Fading · 34%

ML Engineer - Remote

≈ $183k – $406k per year (Estimated) · New York · Remote (United States) · Master's Degree
Python
AI/ML
llama.cpp
vLLM
JAX
SciPy
SGLang
Transformers
NumPy
Hugging Face
PyTorch
Machine Learning
+3
Apply

Clariti

clariticloud.com
clariticloud.com · HQ: Vancouver, Canada · Digital Government · 51-200 employees · Est. 1985 · Hiring 11
HQ: Vancouver, Canada · Digital Government · 51-200 employees · Est. 1985 · Hiring 11
Clariti builds community development software for cities and counties, covering permitting, licensing and plan review workflows for local governments. It is adding AI to its products and hires software engineers, engagement managers, professional services directors and product designers. The company is headquartered in Vancouver, Canada.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago 13 days ago Hiring now · 90%

Senior Applied AI Engineer

≈ $103k – $198k per year (Estimated) · Senior · 7+ years exp · Remote (Canada) · Full-Time · Senior
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
Function Calling
Diffusion Models
NLP
LLM Guardrails
Interpretability
Model Distillation
TensorFlow
Computer Vision
PyTorch
LLM
Machine Learning
AI Agents
DevOps
Git
GCP
Azure
CI/CD
AWS
+13
Apply

NTT DATA

nttdata.com
nttdata.com · HQ: Tokyo, Toyosu, Japan · Health Care · Cybersecurity · Data Centers & Colocation · Information Technology · Managed IT Services (MSP) · Financial Software & Embedded Finance · IT Consulting & Digital Transformation · Systems Integrators · 5000+ employees · Est. 1988 · Trust A · Hiring 1786, 255 new
HQ: Tokyo, Toyosu, Japan · Health Care · Cybersecurity · Data Centers & Colocation · Information Technology · Managed IT Services (MSP) · Financial Software & Embedded Finance · IT Consulting & Digital Transformation · Systems Integrators · 5000+ employees · Est. 1988 · Trust A · Hiring 1786, 255 new
NTT DATA is a Tokyo system integrator founded in 1988 that spun out of the data communications arm of NTT. It is one of the largest information technology services firms in the world, building banking, public sector and payment systems and operating global consulting and managed services. The company runs the CAFIS payment network used across Japanese retail and banking.
Abhijit Dubey · 代表取締役社長 · decision-maker
No careers mailbox known
Verified live · 3 hours ago 13 days ago Likely open · 62%

AI Engineer

Middle · 3+ years exp · Sibiu · Remote (Romania) · Bachelor's Degree · Full-Time · Middle
SQL
Python
Databases
Weaviate
Chroma
pgvector
Pinecone
PostgreSQL
AI/ML
LangGraph
LlamaIndex
Fine-tuning
Embeddings
Scikit-learn
JAX
Function Calling
Chain-of-Thought
Google ADK
Tokenization
Human-in-the-Loop
Multi-Agent Systems
TensorFlow
LangChain
Prompt Engineering
PyTorch
RAG
LLM
AI Agents
Vertex AI
Gemini
Semantic Search
LLM Evaluation
DevOps
gRPC
Rest API
Git
GCP
Azure
CI/CD
AWS
OpenTelemetry
Datadog
Prometheus
Grafana
+33
Apply

Digital Harbor Foundation

digitalharbor.org
digitalharbor.org · HQ: Baltimore, United States · 11-50 employees · Est. 2011 · Hiring 1
HQ: Baltimore, United States · 11-50 employees · Est. 2011 · Hiring 1
Founded in 2011, the Digital Harbor Foundation is a non-profit organization dedicated to promoting digital equity, innovative education, and technology for the public good. Headquartered in Baltimore, Maryland, the foundation provides youth programs, makerspace learning, and resources to bridge the digital divide. Through its strategic initiatives and community projects, it empowers local residents and supports mission-driven tech ventures across the country.
No named decision-maker yet.
No careers mailbox known
Verified live · 21 min ago 14 days ago Fading · 32%

Research Scientist - Digital Harbor Foundation

$120k – $150k per year · Remote (likely United States) · PhD · English (Optional)
Python
Python
Dask
AI/ML
JAX
PyTorch
Machine Learning
Apply
Report

Cohere

cohere.com
cohere.com · HQ: Toronto, Canada · LLM & Generative AI · Foundation Models · AI Agents · 1001-5000 employees · Est. 2019 · Trust A · Hiring 113, 8 new
HQ: Toronto, Canada · LLM & Generative AI · Foundation Models · AI Agents · 1001-5000 employees · Est. 2019 · Trust A · Hiring 113, 8 new
Cohere is a Canadian artificial intelligence company founded in Toronto in 2019 by researchers including one of the co-authors of the transformer architecture paper. It builds large language models aimed specifically at enterprise use rather than consumer chat, with a product line covering text generation, embeddings, reranking and an agent platform, and it emphasises private deployment inside a customer's own cloud or data centre for regulated industries. The company has raised substantial funding from technology and industrial investors, and partners with cloud providers and system integrators to reach banking, healthcare and public sector customers.
Aidan Gomez · Co-founder and Chief Executive Officer · decision-maker
Paul Oran · Executive Officer, Director · decision-maker
Ivan Zhang · Co-founder · founder
Nick Frosst · Co-founder · founder
Careers mailbox is known
Verified live · 2 hours ago 14 days ago Hiring now · 90%

Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)

≈ $235k – $538k per year (Estimated) · Staff+ · New York · Toronto · Remote (EU, Europe, United States, Canada) · Full-Time · Staff
RLHF
Reinforcement Learning
JAX
PPO
GRPO
Post-training
Pre-training
Megatron-LM
FSDP
PyTorch
+2
Apply
Verified live · 2 hours ago 2 months ago Fading · 30%

Member of Technical Staff, Agentic Environments

≈ $235k – $538k per year (Estimated) · Staff+ · New York · Remote (EU, Europe, United States, Canada) · Full-Time · Staff
Python
AI/ML
JAX
Multimodal AI
Cohere SDK
Ray
MLIR
XLA
PyTorch
LLM
DevOps
Kubernetes
+2
Apply
Verified live · 2 hours ago Visa sponsor 5 months ago Long shot · 9%

Machine Learning Intern/Co-op (Winter 2027)

Intern · Remote (United States, United Kingdom, Canada) · Internship · Intern
Python
AI/ML
JAX
NLP
Transformers
CUDA
TPU
MLIR
XLA
TensorFlow
Machine Learning
+2
Apply
Machines can read this selection · checked 10:25 UTC
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.