Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

JAX Jobs with Learning Budget in AI

Browse 2 JAX jobs with Learning Budget in AI for Aug 2026 with roles open at Latitude AI and Cohere. Work as a Senior Software Engineering Manager - Multi-Object Tracking & State Estimation and Member of Technical Staff, Agentic Environments .

JAX is often requested for Research Engineer roles.

Receive emails of JAX Jobs with Learning Budget in AI

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

Filter by benefits

🚑 Medical Insurance 🦷 Dental Insurance 🤓 Vision Insurance 💰 401k 💰 401k Matching 💰 Equity 🏖 Unlimited Vacation 🏖 Paid Time Off 🏔 Company Retreats 🖥 Home Office Budget 📚 Learning Budget 🧘 Mental Wellness Budget ✈️ Visa Sponsorship 📦 Relocation 🏥 HSA

Cities

San Francisco 12 Palo Alto 1 Cambridge 1 Glen Burnie 1 New York 1

Countries

United States 16 California 13 Massachusetts 1 Maryland 1 Canada 1 New York 1

Filter by type

Remote Full Time

C

Member of Technical Staff, Agentic Environments

Cohere / Canada ; New York, NY / Remote / Staff

Python PyTorch JAX

3D

LA

Senior Software Engineering Manager - Multi-Object Tracking & State Estimation

Latitude AI / Remote / Senior

Python PyTorch TensorFlow

$253K – $380K 6D

Frequently asked questions

Which companies are hiring for JAX jobs with Learning Budget in AI?

Top companies hiring include Latitude AI and Cohere.

Are there remote JAX jobs with Learning Budget in AI?

Yes, 2 of 2 are remote.