ZipRecruiter

Log In

1

Pytorch Jobs in Austin, TX (NOW HIRING)

K-Tek Resourcing LLC

Sr. Machine Learning Engineer

Austin, TX · On-site

$113K - $136K/yr

Implement and fine-tune algorithms using frameworks such as PyTorch, TensorFlow, and Scikit-learn. * Develop and maintain production-ready ML solutions and APIs. * Collaborate with data scientists ...

NVIDIA

Senior ML Compiler Engineer

Austin, TX · On-site

$103K - $142K/yr

PyTorch, JAX etc) • Strong python and C/C++ programming skills Preferred : • MS or PhD are preferred • Expertise in AI frameworks such as PyTorch, TensorFlow, and ONNX • Expertise in machine ...

New

Tech Tammina

AIML Engineer (Gen AI)

Austin, TX · On-site

Familiarity with modern ML frameworks (e.g., PyTorch, TensorFlow, scikit-learn) and data tooling. * Strong communication skills to collaborate across cross-functional teams. * Available for ...

Apolis

AIML (Gen AI) - Claud and Python

Austin, TX · On-site

$49.75 - $68.50/hr

Familiarity with modern ML frameworks (e.g., PyTorch, TensorFlow, scikit-learn) and data tooling. * Strong communication skills to collaborate across cross-functional teams. * Available for ...

ClifyX

Python GenAI Model Evaluator

Austin, TX · On-site

$78/hr

PyTorch, TensorFlow, HuggingFace, LangChain. * Evaluation/annotation tools: Scale AI, GroundTruth, Labelbox, Prodigy. * Prompt testing tools: Weights & Biases, MLflow, OpenAI evals, LLM-as-a-judge ...

Bumble Inc.

Staff Machine Learning Engineer

Austin, TX · On-site

PyTorch), ensuring performance, scalability, and reliability in production • Lead the development of LLM-powered systems and AI agents, including prompt design, fine-tuning, and system integration ...

AMD

Software Solutions Architect

Austin, TX · On-site

... PyTorch, inference runtimes) • Apply a data-driven approach to profiling, benchmarking, and tuning • Identify gaps in the software stack and partner with internal teams to close them • Partner ...

Vkore Solutions

GenAI Engineer-W2

Austin, TX · On-site

Train and fine-tune LLMs using PyTorch, DeepSpeed, and LoRA. * Optimize inference using ONNX, vLLM, TensorRT, and GPU acceleration. * Manage datasets, preprocess data, and implement RAG with vector ...

next page

Showing results 1-20

All JobsPytorch JobsPytorch Jobs in Austin, TX

Pytorch information

What are the key skills and qualifications needed to thrive in the Pytorch position, and why are they important?

To thrive in a PyTorch developer role, you need a strong background in deep learning, programming (especially Python), and a solid understanding of machine learning fundamentals, often supported by a degree in computer science, engineering, or a related field. Experience with PyTorch, CUDA, cloud platforms (like AWS or Azure), and familiarity with data processing pipelines are highly valued, and certifications in AI or machine learning can be beneficial. Key soft skills include problem-solving, teamwork, and effective communication to collaborate with cross-functional teams and present technical results clearly. These skills are crucial for building robust machine learning models, ensuring reproducibility, and driving innovation in fast-paced, data-driven environments.

What kinds of projects or tasks can a PyTorch developer expect to work on in a typical role?

As a PyTorch developer, you will likely work on developing, refining, and deploying deep learning models for tasks such as image recognition, natural language processing, or recommendation systems, depending on your company's focus. Your responsibilities may include data preprocessing, model architecture design, experimentation, performance tuning, and collaborating with data scientists and software engineers to integrate models into production systems. You might also be called upon to conduct research or prototype new algorithms, keeping up with the latest advancements in the AI field. Projects can vary from quick proofs of concept to large-scale deployments, offering diverse opportunities to grow your technical and collaborative skills.

What is a PyTorch job?

A PyTorch job typically involves working with the PyTorch deep learning framework to develop, train, and deploy machine learning models. Professionals in this role may build neural networks, perform data preprocessing, optimize models, and integrate them into applications. These jobs are commonly found in AI research, software development, and data science, requiring expertise in Python, deep learning, and model optimization techniques.

What are popular job titles related to Pytorch jobs in Austin, TX? For Pytorch jobs in Austin, TX, the most frequently searched job titles are:
What job categories do people searching Pytorch jobs in Austin, TX look for? The top searched job categories for Pytorch jobs in Austin, TX are:
What cities near Austin, TX are hiring for Pytorch jobs? Cities near Austin, TX with the most Pytorch job openings:
Pytorch jobs near you
Infographic showing various Pytorch job openings in Austin, TX as of June 2026, with employment types broken down into 57% Full Time, 6% Part Time, 6% Temporary, and 31% Contract. Highlights an 75% In-person, and 25% Remote job distribution.
Compiler Engineer - MLIR / PyTorch Infrastructure

Compiler Engineer - MLIR / PyTorch Infrastructure

Mythic

Austin, TX

$106K - $139K/yr

Full-time

Posted 27 days ago

Job description

About us
Mythic is building the future of AI computing with breakthrough analog technology that delivers 100 the performance of traditional digital systems at the same power and cost. This unlocks bigger, more capable models and faster, more responsive applications-whether in edge devices like drones, robotics, and sensors, or in cloud and data center environments. Our technology powers everything from large language models and CNNs to advanced signal processing, and is engineered to operate from -40 C to +125 C, making it ideal for industrial, automotive, aerospace, and defense.

We've raised over $100M from world-class investors including Softbank, Threshold Ventures, Lux Capital, and DCVC, and secured multi-million-dollar customer contracts across multiple markets.

About the role
Join us in advancing the MLIR ecosystem at Mythic. You'll help extend our existing high-level dialects and design a new hardware-aware low-level dialect, building conversion paths that bridge to our current IRs. Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an MLIR-based stack that unifies hardware constraints with developer needs and accelerates adoption of modern ecosystems.
Here's what you will do
  • Architect the migration of the existing compiler flow into MLIR, defining dialects, passes, and lowering strategies.
  • Build conversion paths between MLIR and Mythic's custom low-level IR to keep both flows operational during migration.
  • Define validation infrastructure within MLIR, including interpretation or execution paths for simulation and debugging.
  • Enable compilation by extending MLIR integration across analog accelerators and digital subsystems.
  • Leverage Torch-MLIR where PyTorch inputs are available, and guide future integration with PyTorch 2.0 compiler technologies (TorchInductor, TorchDynamo, Torch-MLIR)
Here's the background we hope you will have
  • 3+ years of experience in compiler or high-performance systems development.
  • Proficiency in modern C++ (C++14/17/20) and Python.
  • Direct, hands-on experience with MLIR, including dialect design, compiler passes, or lowering pipelines.
  • Strong understanding of compiler IRs and transformations, with the ability to reason about lowering from high-level ops to hardware-aware representations.
The following would be nice to have, but is not required
  • Experience architecting complete MLIR flows: from frontend dialects down to hardware-aware dialects, including conversion to and from existing IRs.
  • Familiarity with PyTorch compiler technologies, especially Torch-MLIR and integration paths with PyTorch 2.0 (TorchDynamo, TorchInductor).
  • Knowledge of dataflow architectures, scheduling, and memory orchestration.
  • Background in heterogeneous or specialized accelerators (e.g., analog compute, NPUs, GPUs, DSPs).
What we offer
  • Shape the MLIR and PyTorch integration strategy for novel AI hardware.
  • Opportunity to contribute upstream and influence the broader ML compiler ecosystem.
  • Collaborate with a team spanning hardware, runtime, and compiler design.
  • Competitive compensation, equity, and comprehensive benefits.
At Mythic, we foster a collaborative and respectful environment where people can do their best work. We hire smart, capable individuals, provide the tools and support they need, and trust them to deliver. Our team brings a wide range of experiences and perspectives, which we see as a strength in solving hard problems together. We value professionalism, creativity, and integrity, and strive to make Mythic a place where every employee feels they belong and can contribute meaningfully.
apply for this job

Apply