Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

JAX Jobs with Learning Budget in AI

Browse 3 JAX jobs with Learning Budget in AI for Oct 2026 with roles open at Planet Labs Inc., Cohere, and Crusoe. Work as a Senior Software Engineer, Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization), and Senior Staff Software Engineer, Storage.

JAX is often requested for Software Engineer jobs 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 Full Stack Engineer HR JAX MongoDB NCCL OpenTelemetry Platform Engineer Product Manager Program Manager RDMA React Rust SEO Snowflake Tableau 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 💊 FSA 🏥 HSA

Cities

San Francisco 11 Seattle 3 Santa Clara 2 Amsterdam 2 Toronto 2 Cambridge 2

Countries

United States 21 California 15 Washington 3 Netherlands 2 Canada 2 United Kingdom 2

Filter by type

Remote Full Time Internship

C

Senior Staff Software Engineer, Storage

Crusoe / Sunnyvale, United States / Staff

PyTorch TensorFlow JAX

$245K – $290K 1D

C

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

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

PyTorch JAX RLHF

2D

PLI

Senior Software Engineer

Planet Labs Inc. / Ljubljana, Slovenia / Senior

€58K – €72K 9D

Frequently asked questions

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

Top companies hiring include Planet Labs Inc., Cohere, and Crusoe.

Are there remote JAX jobs with Learning Budget in AI?

Yes, 1 of 3 are remote.