Direct from source · No middlemen

Jax In Jobs

110 open positions · Updated 1 week ago

Average salary (USD/year): 218.9k–324.6k/yr

Showing 20 of 110 positions

Search with filters →
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 2 months ago
Flexible on stack
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 1 week 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 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 5 months 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 10 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 2 days 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
Yuno

Join Yuno as a Senior AI Engineer to architect and scale AI-powered systems that redefine payment orchestration.

Yuno Hyderabad Published 5 months ago
Flexible on stack
Fireworks AI

Join Fireworks AI as a Member of Technical Staff to advance generative AI through foundational research and collaboration with top experts.

Fireworks AI San Mateo Published 1 month 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 5 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 4 months ago
Flexible on stack
Perplexity AI

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

Perplexity AI London Published 5 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 Belgrade Published 8 months ago
Flexible on stack
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 5 months ago
Flexible on stack
AssemblyAI

Join AssemblyAI as a Senior Research Engineer to enhance large-scale distributed training and inference systems in Voice AI.

AssemblyAI Remote - New York $270k–$310k/yr Published 1 week 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
Page 1 of 6 Next →