"jax in" Jobs

110 open tech roles matching “jax in”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: JAX, PyTorch, Python. Every listing is re-checked daily and closed roles are removed.

Showing 20 of 110 results

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 3 months ago
Flexible on stack
World Labs

Join World Labs as a Performance Engineer to optimize AI models for speed and efficiency in a cutting-edge research environment.

World Labs San Francisco $200k–$300k/yr Published 4 months ago
Flexible on stack 70% coding
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
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 2 weeks 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 6 days 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 8 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 7 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 6 days 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 2 months ago
Flexible on stack
Perplexity AI

Join Perplexity AI as a Technical Staff member to enhance our AI inference engine with cutting-edge technologies.

Perplexity AI San Francisco Published 5 months ago
Flexible on stack
Anthropic
Anthropic San Francisco, CA $315k–$560k/yr Published 10 months ago
Block

Lead the development of machine learning initiatives to enhance customer support experiences at Block.

Block Seattle, WA, United States of America $276.8k–$415.2k/yr Published 3 months ago
Flexible on stack
P-1 AI

Join P-1 AI as an AI Research Scientist to push the boundaries of AI applied to the physical world in a hands-on role.

P-1 AI San Francisco Bay Area Published 1 month ago
Flexible on stack
Reflection AI

Join Reflection AI as a Research Software Engineer to bridge research and production in cutting-edge AI training systems.

Reflection AI New York, NY Published 6 months ago
Flexible on stack
ChipAgents

Join ChipAgents as a Research Scientist to advance AI-assisted electronic design automation in a high-impact, mission-driven team.

ChipAgents San Jose $150k–$350k/yr Published 3 months ago
Flexible on stack
Discord

Join Discord as a Senior Machine Learning Engineer to build and deploy models that enhance user safety on the platform.

Discord San Francisco, California, United States $220k–$247.5k/yr Published 1 day ago
Flexible on stack