Triton Jobs

The Core ML team is seeking an engineer to map novel algorithms for efficient large-scale AI training and inference to the Cerebras architecture. This role involves determining when these algorithms outperform competing approaches and how their advantages scale with models, workloads, and hardware. You will combine analytical performance modeling, empirical benchmarking, and hands-on prototyping to characterize the efficiency frontiers of emerging ML algorithms. Your work will span kernel-level and end-to-end performance, influencing research directions, implementation on current systems, and future hardware/software capabilities.

The Core ML team is seeking an engineer to bridge the gap between novel machine learning research and efficient execution on Cerebras systems. You will work across ML frameworks, compilers, runtimes, and low-level kernels to implement new algorithmic capabilities, diagnose performance bottlenecks, and transform research prototypes into robust, high-performance demonstrations. Depending on your background, your work may emphasize runtime capabilities such as token orchestration, scheduling, communication, and distributed execution; low-level kernel development for novel ML operations; or a combination of both. This role is crucial for advancing the capabilities of Cerebras' unique AI chip architecture.

Mirage is seeking an ML Engineer to build and scale the systems powering their video generation models. This role sits at the intersection of research and systems engineering, focusing on making advanced models faster, more efficient, and capable of ultra-low latency, real-time generation. You will work on novel modeling approaches, training objectives, scaling strategies, and inference optimization and efficiency to bring cutting-edge models into production.

$175k - $300k

Together AI

Together AI is seeking a Machine Learning Engineer to join our Inference Engine team, focusing on optimizing and enhancing the performance of our AI inference systems. This role involves working with state-of-the-art large language models and ensuring they run efficiently and effectively at scale. You will have the chance to collaborate closely with AI researchers and engineers to create cutting-edge AI solutions and shape the future of AI inference.

$200k - $300k

Together AI

As a Systems Research Engineer specialized in GPU Programming, you will play a crucial role in developing and optimizing GPU-accelerated kernels and algorithms for ML/AI applications. You will co-design GPU kernels and model architecture to enhance the performance and efficiency of our AI systems, and contribute to the co-design of efficient GPU architectures and programming models. Your research skills will be vital in staying up-to-date with the latest advancements in GPU programming techniques, ensuring that our AI infrastructure remains at the forefront of innovation.

$200k - $300k

Together AI

As a Systems Research Engineer Intern specialized in GPU Programming, you will play a crucial role in developing and optimizing GPU-accelerated kernels and algorithms for ML/AI applications. You will co-design GPU kernels and model architecture to enhance the performance and efficiency of our AI systems, and contribute to the co-design of efficient GPU architectures and programming models. Your research skills will be vital in staying up-to-date with the latest advancements in GPU programming techniques, ensuring that our AI infrastructure remains at the forefront of innovation. This internship is based on-site at our San Francisco HQ, running through the Winter term from January to April.

Together AI

As a Systems Research Engineer Intern specialized in GPU Programming, you will play a crucial role in developing and optimizing GPU-accelerated kernels and algorithms for ML/AI applications. You will co-design GPU kernels and model architecture to enhance the performance and efficiency of our AI systems, and contribute to the co-design of efficient GPU architectures and programming models. Your research skills will be vital in staying up-to-date with the latest advancements in GPU programming techniques, ensuring that our AI infrastructure remains at the forefront of innovation.

$121k - $146k

Together AI

The Model Shaping team at Together AI focuses on tailoring open foundation models for downstream applications, building services that help machine learning developers select and improve models with domain-specific data. We also develop novel methods for efficient model training and evaluation, drawing from machine learning, natural language processing, and ML systems. As a Research Engineer on the Scaling Team, you will translate cutting-edge research in efficient foundation model training into robust, high-performance systems. You will optimize Together's training infrastructure, identify performance bottlenecks, and implement state-of-the-art techniques in production environments. Your work will directly influence the fine-tuning experience for Together's customers. You will integrate newly released open-source models onto the platform, ensuring efficient and reliable training across diverse workloads. Collaborating with Research Scientists, you will build experimental infrastructure to accelerate research and deploy validated ideas at scale.

$200k - $290k

NVIDIA is seeking versatile software engineers for its XLA team to build high-performance, production-grade software at the core of next-generation AI systems. This role involves developing compiler optimization algorithms for deep learning workloads, optimizing inference and training performance for the JAX framework and the OpenXLA compiler on NVIDIA GPUs at scale. You will collaborate with deep learning framework and hardware architecture teams to accelerate future AI software.

$152k - $288k

NVIDIA AI for Media is seeking outstanding engineers to join its team, focusing on building and optimizing real-time AI solutions for media and entertainment workflows. This role will involve setting the technical direction for content authenticity AI, architecting model and pipeline strategies for detecting synthetic and manipulated media, and making critical build-vs-adapt decisions. The team works on core technologies to solve ambitious computer vision and deep learning problems, aiming to deliver studio-quality audio and high-resolution video enhancement and effects.

$272k - $431k

NVIDIA is seeking an NCX Senior Engineer to join our DSX team, collaborating closely with strategic customers to implement and enhance groundbreaking AI workloads. You will deliver hands-on technical assistance for advanced AI deployments, intricate distributed systems, and ensure customers realize efficient performance from NVIDIA's AI platform across varied environments. We partner with the world's most innovative AI companies to address their most challenging technical problems. In this role, you will develop innovative solutions that advance AI infrastructure capabilities and directly influence customer success with breakthrough AI initiatives.

$184k - $357k

NVIDIA is seeking a motivated Deep Learning engineer to integrate advanced CUDA features and Distributed Runtime technologies into AI stacks, including frameworks like PyTorch, TRT-LLM, vLLM, SGLang, and JAX. You will join the team responsible for core CUDA features and runtimes that scale Deep Learning and HPC applications, addressing diverse multi-GPU demands from large-scale training to low-latency inference. Your work will enhance the productivity and performance of AI applications, accelerating community adoption through AI toolkits. This role offers a significant opportunity to advance the state of the art in AI technologies.

$184k - $357k