"jax" Jobs

Join DeepL as a Senior Research Scientist to innovate in reinforcement learning and shape the future of AI technology.

Join SpaceX as a Senior Software Engineer to develop AI systems that enhance engineering processes for launch vehicles and spacecraft.

Join Armada as an AI Engineer to build and deploy intelligent systems at the edge of the physical world.

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 SpaceX as an AI Software Engineer to develop AI systems that enhance engineering processes for launch vehicles and spacecraft.

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 Cloudflare as a Senior Machine Learning Engineer to optimize and productionize ML models for a global serverless inference platform.

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

Join CuspAI as an Applied AI/ML Engineer to design intelligent agents for autonomous materials discovery.

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.

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

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

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

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

Join Proxima Fusion as a Computational Engineer to tackle complex engineering challenges in the development of commercial fusion power plants.