Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

Remote Full Time JAX Jobs in AI

Browse 2 Remote Full Time JAX jobs in AI for Oct 2026 with roles open at Discord and Cohere. Work as a Staff Software Engineer, Machine Learning (Consumer Revenue) and Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization).

JAX is often requested for Software Engineer jobs roles.

Receive emails of Remote Full Time JAX Jobs in AI

AI Engineer Ansible Customer Success Manager Cypress Data Engineer Data Scientist Elasticsearch Engineering Manager Ethernet Full Stack Engineer HR JAX MongoDB NCCL OpenTelemetry Platform Engineer Product Manager Program Manager RDMA React Rust SEO Sketch Tableau Triton TypeScript

Cities

San Francisco 1 New York 1 Toronto 1

Countries

United States 2 California 1 Canada 1 New York 1

Filter by type

Remote Full Time Internship

C

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

Cohere / New York, NY ; Toronto, CA / Remote / Staff

PyTorch JAX RLHF

2D

D

Staff Software Engineer, Machine Learning (Consumer Revenue)

Discord / San Francisco, United States / Remote / Staff

Python PyTorch TensorFlow

$272K – $340K 25D

Frequently asked questions

Which companies are hiring for Remote Full Time JAX jobs in AI?

Top companies hiring include Discord and Cohere.