"jax" Jobs

129 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 Berlin right now — showing "jax" jobs in all locations.

Showing 20 of 129 results

Periodic Labs

Join Periodic Labs as a Computational Scientist to develop differentiable simulations for continuum-physics problems using AI.

Periodic Labs Menlo Park, CA $250k–$350k/yr Published 2 months ago
Flexible on stack
Perplexity AI

Join Perplexity AI as a Search Machine Learning Engineer Intern to enhance search technologies in a collaborative environment.

Perplexity AI London Published 7 months ago
Flexible on stack
Inferact

Join Inferact as a co-op student to work on cutting-edge AI inference systems in a hands-on engineering role.

Inferact San Francisco Published 1 month ago
Flexible on stack
Eka Robotics

Join Eka Robotics as a Machine Learning Engineer to develop cutting-edge algorithms for robotic manipulation in a fast-paced startup environment.

Eka Robotics Boston Area Published 1 year ago
Flexible on stack
Udio

Lead the unification of large datasets to power generative audio models in a fast-moving team at Udio.

Udio New York $180k–$220k/yr Published 12 months ago
Flexible on stack
Inferact

Join Inferact as a TPU performance engineer to optimize vLLM for Google TPUs, enhancing AI inference performance.

Inferact San Francisco $200k–$400k/yr Published 4 months ago
Flexible on stack
Inferact

Join Inferact as a TPU performance engineer to optimize vLLM for Google TPUs, enhancing AI inference performance.

Inferact Singapore S$200k–S$400k/yr Published 4 months ago
Flexible on stack
Eka Robotics

Join Eka Robotics as a Machine Learning Engineer to develop cutting-edge computer vision systems for robotic manipulation.

Eka Robotics Boston Area Published 1 year ago
Flexible on stack
Perplexity AI

Join Perplexity AI as a Search Machine Learning Engineer Intern to enhance search technologies in a collaborative environment.

Perplexity AI Belgrade Published 10 months ago
Flexible on stack
Coreweave

Join CoreWeave as a Senior Applied ML Engineer to tackle challenges in continuous learning for AI agents with a focus on innovative solutions.

Coreweave Bellevue, WA / Sunnyvale, CA $182k–$242k/yr Published 7 months ago
Flexible on stack
Eka Robotics

Own and improve the development experience by designing and maintaining build systems and developer tooling at Eka Robotics.

Eka Robotics Boston Area Published 4 weeks ago
Flexible on stack
Applied Intuition

Join Applied Intuition as an ML Runtime Optimization Engineer to optimize ML models for embedded environments in a collaborative team.

Applied Intuition Sunnyvale Published 1 year ago
Flexible on stack
Atoms

Join Atoms as an AI Research Intern to bridge AI research and real-world applications in autonomous transport platforms.

Atoms Mountain View, CA $70–$70/hr Published 2 weeks ago
Flexible on stack
Etched

Join Etched as an Inference Software Engineer to build and optimize cutting-edge AI inference systems in a fully in-person team.

Etched San Jose Published 1 year ago
Flexible on stack
Cursor

Join SpaceXSI as a Sr. Model Engineer to drive research in reinforcement learning and improve model capabilities.

Cursor Palo Alto, CA $220k–$357k/yr Published 2 days ago
Flexible on stack
Etched

Join Etched as an Inference Intern to work on next-generation AI accelerators in a hands-on, collaborative environment.

Etched San Jose Published 11 months ago
Flexible on stack
Periodic Labs

Join Periodic Labs as an ML Systems Engineer to build and optimize large-scale training and reinforcement learning infrastructure.

Periodic Labs Menlo Park, CA $250k–$350k/yr Published 6 months ago
Flexible on stack