"jax" Jobs
23 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: JAX, Python, PyTorch. Every listing is re-checked daily and closed roles are removed.
Showing 20 of 23 results
Join Inferact as a performance engineer to optimize vLLM, the fastest AI inference engine, working directly with hardware vendors.
Join Fireworks AI as a Member of Technical Staff to advance generative AI through foundational research and collaboration with top experts.
Own the strategy and execution for infrastructure-native products at the intersection of W&B and CoreWeave.
Join Perplexity AI as an AI Inference Engineer to optimize and develop our inference engine for various model architectures.
Join Reflection AI as a Member of Technical Staff to design and implement data sourcing and curation techniques for AI models.
Join Harvey as a Research Engineer to build infrastructure for ML models in a fast-paced, fully remote environment.
Join Udio as a full-stack scientist to lead quantitative research efforts at the intersection of music and AI.
Join Harvey AI as a Member of Technical Staff to build infrastructure for ML models in legal work.
Join Perplexity AI as a Technical Staff member to enhance our AI inference engine with cutting-edge technologies.
Join Inductive Bio as a Machine Learning expert to innovate on drug discovery models and collaborate with scientists.
Join Discord as a Staff Machine Learning Engineer to build ML systems for consumer revenue across various platforms.
Join Arena as a Machine Learning Scientist to evaluate AI models and contribute to impactful research in a collaborative environment.
Related searches