"jax" Jobs

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

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

Join Baseten as a Forward Deployed Engineer to solve complex AI challenges for leading companies.

Join Xaira Therapeutics as a Biomedical AI Engineer to design AI systems that accelerate therapeutic discovery.

Join Xaira Therapeutics as an AI Scientist to develop multimodal foundation models for drug discovery using advanced AI techniques.

Join Goodfire as a Forward Deployed Engineer to integrate AI interpretability solutions with strategic partners.

Drive the development of AI-powered clinician tools at Abridge, combining clinical expertise with AI in a hybrid work environment.

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

Join P-1 AI as an AI Research Scientist to advance spatial reasoning and physical world modeling in cutting-edge AI systems.

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

Lead the capacity management for Baseten's TPU fleet, ensuring optimal performance and reliability in AI workloads.

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

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

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

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

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

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.

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