Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

Remote Full Time JAX AI Jobs in United States

Browse 2 AI jobs in United States 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 AI Jobs in United States

AI Engineer Android Engineer Ansible Cypress Data Engineer Data Scientist Elasticsearch Engineering Manager Ethernet Full Stack Engineer HR JAX MongoDB OpenTelemetry Platform Engineer Product Manager Program Manager React Reinforcement Learning Rust SEO Sketch Snowflake Spark Tableau TypeScript

Cities

San Francisco 1 New York 1 Toronto 1

Countries

United States 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 24D

Frequently asked questions

Which companies are hiring for Remote Full Time JAX jobs?

Top companies hiring include Discord and Cohere.