Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

Remote JAX Jobs in AI

Browse 2 Remote 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 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 OpenTelemetry Platform Engineer Product Manager Program Manager RDMA React Rust SEO Sketch Snowflake 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

3D

D

Staff Software Engineer, Machine Learning (Consumer Revenue)

Discord / San Francisco, United States / Remote / Staff

Python PyTorch TensorFlow

$272K – $340K 26D

Frequently asked questions

Which companies are hiring for Remote JAX jobs in AI?

Top companies hiring include Discord and Cohere.