"jax" Jobs

116 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Every listing is re-checked daily and closed roles are removed.

No "jax" jobs in In San Francisco right now — showing "jax" jobs in all locations.

Showing 20 of 116 results

Atoms

Join Atoms as a Staff Machine Learning Engineer to bridge AI research and real-world physical actuation in autonomous transport.

Atoms San Francisco, CA $273k–$345k/yr Published 3 months ago
Flexible on stack 70% coding
Perplexity AI

Join Perplexity AI as an AI Inference Engineer to optimize and develop our inference engine for various model architectures.

Perplexity AI London Published 5 months ago
Flexible on stack
Block

Join Block as an Applied Research Intern to develop proactive intelligence systems that enhance customer understanding and decision-making.

Block Bay Area, CA, United States of America Published 3 months ago
Flexible on stack
Block

Join Block as an Applied Research Intern to develop proactive intelligence systems that enhance customer understanding and decision-making.

Block Toronto, Ontario , Canada Published 3 months ago
Flexible on stack
Proxima Fusion

Join Proxima Fusion as a Software Engineer to develop next-generation simulation tools for stellarator design and research.

Proxima Fusion Munich Published 1 week ago
Flexible on stack
Proxima Fusion

Join Proxima Fusion as a Multiphysics Simulation Engineer to shape the architecture of the world's first commercial fusion power plant.

Proxima Fusion Munich Published 1 week ago
Flexible on stack
Chai Discovery

Join Chai Discovery as a Research Engineer to optimize ML infrastructure for drug discovery in a fast-paced, innovative environment.

Chai Discovery San Francisco, United States Published 3 days ago
Flexible on stack
Graphcore

Join Graphcore's Triton team to enhance AI frameworks and optimize software for cutting-edge machine learning workloads.

Graphcore Gdańsk, Pomeranian Voivodeship, Poland Published 6 months ago
Flexible on stack
Instacart

Lead the design and development of core ML models for Instacart’s ads ecosystem in a fully remote role.

Instacart United States - Remote $201k–$253.5k/yr Published 4 months ago
Flexible on stack
Netic

Join Netic as an Applied AI Research Engineer to drive impactful ML projects in a cutting-edge AI-first environment.

Netic San Francisco Published 1 year ago
Flexible on stack
Together AI

Join Together AI as a Research Intern to work on cutting-edge agentic AI systems in a collaborative environment.

Together AI San Francisco $58–$70/hr Published 1 week ago
Flexible on stack
Together AI

Join Together AI as a Research Intern to explore frontier AI systems and contribute to innovative projects in agentic AI.

Together AI San Francisco $58–$70/hr Published 1 week ago
Flexible on stack
Graphcore

Join Graphcore's Triton team to enhance AI frameworks and improve performance on cutting-edge hardware.

Graphcore Bristol, UK; Gdańsk, Pomeranian Voivodeship, Poland Published 1 month ago
Flexible on stack
Dialpad

Join Dialpad as a Senior Software Engineer to build and improve the AI/ML inference platform for enterprise-scale applications.

Dialpad Buenos Aires, Argentina Published 2 weeks ago
Flexible on stack 70% coding
Ideogram

Join Ideogram as an Applied ML Engineer to bridge research and product, turning generative models into production features.

Ideogram Toronto Published 2 months ago
Flexible on stack
Atomic AI

Join Atomic AI as a Scientist to develop machine learning models for RNA structure prediction and drug targeting.

Atomic AI South San Francisco, California, United States $170k–$220k/yr Published 9 months ago
Flexible on stack
Xaira Therapeutics

Join Xaira Therapeutics as a Senior Software Engineer to build AI infrastructure for drug discovery and development.

Xaira Therapeutics Seattle, Washington, United States $189k–$245k/yr Published 3 months ago
Flexible on stack
Udio

Join Udio as a full-stack scientist to lead quantitative research efforts at the intersection of music and AI.

Udio New York City (Remote possible for exceptional candidates) $250k–$350k/yr Published 8 months ago
Flexible on stack
Cantina

Join Cantina as a Research Intern to work on next-gen video models in a hands-on, mentorship-driven environment.

Cantina Singapore Published 2 weeks ago
Flexible on stack
Twilio

Join Twilio as a Machine Learning Engineer to design AI-powered features that enhance customer conversations.

Twilio Remote - Spain Published 3 months ago
Flexible on stack