Machine Learning Engineer Jobs in Austin

Join the GPU Foundations Developer Tools team at NVIDIA and contribute to the development of groundbreaking performance analysis tools for GPUs and SoCs. You will innovate as you develop performance analysis capabilities for NVIDIA hardware, drivers, and SDKs, supporting current and future generations of GPUs. Your work will empower inventors worldwide to maximize the potential of our GPU hardware in their applications, directly impacting the performance profiling tools based on next-generation hardware that you help define.

$152k - $288k

NVIDIA is seeking a Backend Compiler Engineer to join our GPU Software organization. The Compiler team is responsible for constructing and emitting high-performance GPU machine instructions for Graphics (OpenGL, Vulkan, DX) and Compute (CUDA, PTX, OpenCL, Fortran, C++). This team comprises leading compiler engineering experts who deliver cutting-edge performance and capabilities for NVIDIA's complex parallel SIMT architectures.

$152k - $242k

SambaNova

SambaNova is seeking a Principal Place and Route (PnR) Engineer to join our team. This role involves developing and tuning the software that maps dataflow graphs onto SambaNova's Reconfigurable Dataflow Unit (RDU). You will work on partitioning, placement, and routing algorithms to optimize the mapping of customer workloads, including large language model inference and training, onto our hardware. Your contributions will directly impact the throughput and performance of compiled programs and will be integrated into production releases.

$0k - $1000k

As a Senior AI Performance Engineer, you will analyze and optimize AI training and inference workloads across large-scale distributed systems. Your work will connect compute, memory, communication, and software behavior, owning complex performance investigations from problem definition through validation. You will translate profiling, benchmarking, and modeling results into actionable improvements for engineers. This role offers broad scope across hardware, software, networking, and system architecture, contributing to Graphcore's delivery of efficient and reliable AI systems for large-scale deployment.

Salary not listed

SambaNova

SambaNova is seeking a new grad ML Infrastructure Engineer to join their team. This role involves writing production Python code for the inference serving stack, including vLLM integration and public APIs. You will contribute to bringing features like speculative decoding, prompt caching, and tool calling into production on the RDU, and develop accuracy and regression tests to ensure quality before customer release.

$131k - $197k

SambaNova

SambaNova is seeking a new grad Compiler Engineer to work on core areas of our compiler, which powers the RDU (Reconfigurable Dataflow Unit) chip. This role involves working on the compiler's IR and pass infrastructure, model lowering and partitioning, tensor tiling, memory management, and mapping operations onto the accelerator fabric. You will gain insight into how compiler decisions impact chip performance, write production code in C++ and Python, perform low-level debugging, and collaborate with runtime and hardware teams. Senior engineers will provide code reviews and mentorship.

$131k - $197k

Tenstorrent is seeking engineers to work at the hardware/software boundary, defining how hardware blocks are initialized, coordinated, and used by the runtime. You will determine how workloads are dispatched, scheduled, synchronized, and executed across the accelerator, solving complex performance problems through implementation, bring-up, and debugging. This role involves defining the runtime architecture and implementation that controls execution, scheduling, synchronization, and resource management across Tenstorrent’s custom processor, accelerator, memory, and data-movement IP. You will build low-level runtime components, design runtime interfaces, and analyze bottlenecks across the runtime and hardware/software stack to drive improvements in latency, throughput, scalability, and reliability.

$100k - $500k

Tenstorrent is a leader in AI technology, aiming to revolutionize performance, usability, and cost-efficiency. As AI redefines computing, our solutions must integrate innovations across software models, compilers, platforms, networking, and semiconductors. We are seeking a CPU Performance Modeling Architect to evaluate, shape, and optimize future CPU architectures. This role involves using performance modeling, workload analysis, and a deep understanding of CPU architecture to address complex design questions. You will collaborate closely with CPU architects, RTL designers, software and compiler teams, and system engineers to identify performance opportunities, assess architectural trade-offs, and translate modeling insights into actionable design decisions.

$100k - $500k

Tenstorrent is at the forefront of AI technology, aiming to revolutionize performance, usability, and cost-efficiency. As AI reshapes computing, our solutions must integrate software models, compilers, platforms, networking, and semiconductors. We have developed a high-performance RISC-V CPU and are seeking passionate individuals to build the ultimate AI platform. Our Tensix Team is architecting the future of AI compute with scalable RISC-V processors, focusing on reimagining RISC-V core frontends for enhanced programmability, efficiency, and developer experience. This is a unique opportunity to influence the CPU architecture at the core of our AI platform and lead a critical technical initiative.

Salary not listed

Tenstorrent is seeking a Physical Design Methodology Engineer to develop, deploy, and own the RTL-to-GDSII methodology for their AI hardware IP team. This role is crucial for ensuring that new blocks, nodes, or customer variants can start from a working flow, rather than a cold start. The ideal candidate will be automation-minded, focused on removing manual steps and making PPA exploration repeatable, and will leverage AI, including LLMs and ML-driven optimization, in their flow development. This is a hybrid role based in Toronto, Austin, or Belgrade, and the company welcomes candidates at various experience levels.

$100k - $500k