ZipRecruiter

Log In

1

Pytorch Developer Jobs in Austin, TX (NOW HIRING)

Apolis

AIML (Gen AI) - Claud and Python

Austin, TX · On-site

$49.75 - $68.50/hr

Ability to work independently and integrate with an existing engineering team. * Familiarity with modern ML frameworks (e.g., PyTorch, TensorFlow, scikit-learn) and data tooling. * Strong ...

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 ... Collaborate with data scientists, data engineers, software developers, and business stakeholders to ...

eStaffLLC

Senior Software Developer Specialist

Austin, TX · On-site

$54 - $71.25/hr

Production CV experience with PyTorch/TensorFlow, OpenCV, YOLO, object detection, segmentation, or ... CI/CD: Azure DevOps, GitHub Actions, Jenkins, or comparable automation pipeline experience.

eStaffLLC

Senior Software Developer Specialist

Austin, TX · On-site

$54 - $71.25/hr

Computer Vision Production CV experience with PyTorch/TensorFlow, OpenCV, YOLO, object detection ... Azure DevOps, GitHub Actions, Jenkins, or comparable automation pipeline experience. Feature Stores:

Tech Tammina

AIML Engineer (Gen AI)

Austin, TX · On-site

Role: - AIML Engineer (Gen AI) - keyskill: Claud and Python, ML engineer background Location ... Familiarity with modern ML frameworks (e.g., PyTorch, TensorFlow, scikit-learn) and data tooling.

VBeyond

Python Developer / Engineer

Austin, TX · On-site

$49.75 - $68.50/hr

Role: Python Developer / Engineer Location: San Jose, CA / Costa Mesa, CA / Austin, TX (Hybrid ... TensorFlow, PyTorch, scikit-learn. * Experience with orchestration tools: Kubeflow, MLflow, Airflow ...

eStaffLLC

Senior Software Developer Specialist

Austin, TX · On-site

$54 - $71.25/hr

Production CV experience with PyTorch/TensorFlow, OpenCV, YOLO, object detection, segmentation, or ... CI/CD: Azure DevOps, GitHub Actions, Jenkins, or comparable automation pipeline experience.

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 ...

Bumble Inc.

Staff Machine Learning Engineer

Austin, TX · On-site

... Engineer to lead the development of advanced machine learning systems. In this role, you will be ... PyTorch), ensuring performance, scalability, and reliability in production • Lead the development ...

next page

Showing results 1-20

All JobsPytorch Developer JobsPytorch Developer Jobs in Austin, TX

Pytorch Developer information

What is a PyTorch Developer?

A PyTorch Developer is a software engineer or data scientist who specializes in using PyTorch, an open-source machine learning library, to build and deploy deep learning models. Their responsibilities typically include designing neural network architectures, training and evaluating models, and optimizing code for performance. PyTorch Developers work in fields such as artificial intelligence, computer vision, and natural language processing, collaborating with teams to solve complex problems using machine learning. They are proficient in Python and have a strong understanding of deep learning concepts. Additionally, they often contribute to research, development, and the deployment of AI solutions in production environments.

What are the key skills and qualifications needed to thrive as a Pytorch Developer, and why are they important?

To thrive as a Pytorch Developer, you need strong programming skills in Python, a solid grasp of machine learning concepts, and experience with deep learning frameworks—especially PyTorch itself. Familiarity with tools like CUDA, Jupyter Notebooks, and version control systems (e.g., Git) is typically expected, along with knowledge of cloud platforms or relevant certifications. Problem-solving ability, effective collaboration, and clear communication are crucial soft skills for success in this role. These skills and qualities are vital for efficiently building, optimizing, and deploying machine learning models in real-world applications.

What is the difference between Pytorch Developer vs Machine Learning Engineer?

AspectPytorch DeveloperMachine Learning Engineer
Required CredentialsBachelor's or higher in CS, experience with PyTorchBachelor's or higher in CS, data science, or related field, with ML experience
Work EnvironmentResearch labs, AI startups, tech companies focusing on deep learningTech companies, finance, healthcare, often involving deployment and scaling ML models
Industry UsagePrimarily in AI research and development teamsAcross industries implementing ML solutions in production

While both roles require knowledge of machine learning and experience with PyTorch, a Pytorch Developer mainly focuses on developing and optimizing deep learning models using PyTorch. A Machine Learning Engineer often has a broader scope, including deploying, maintaining, and scaling ML models across various platforms and industries.

What are some common challenges Pytorch Developers face when deploying machine learning models to production environments?

Pytorch Developers often encounter challenges when transitioning models from research to production, such as optimizing model performance for inference speed and memory usage, ensuring compatibility with deployment frameworks like TorchScript or ONNX, and managing dependencies across different systems. Additionally, integrating PyTorch models into existing software stacks and maintaining reproducibility can be complex. Collaborating closely with DevOps and data engineering teams is crucial to address these issues and ensure smooth deployment.
What are popular job titles related to Pytorch Developer jobs in Austin, TX? For Pytorch Developer jobs in Austin, TX, the most frequently searched job titles are:
What job categories do people searching Pytorch Developer jobs in Austin, TX look for? The top searched job categories for Pytorch Developer jobs in Austin, TX are:
What cities near Austin, TX are hiring for Pytorch Developer jobs? Cities near Austin, TX with the most Pytorch Developer job openings:
Pytorch Developer jobs near you
Infographic showing various Pytorch Developer job openings in Austin, TX as of June 2026, with employment types broken down into 85% Full Time, 1% Part Time, and 14% Contract. Highlights an 83% Physical, 4% Hybrid, and 13% Remote job distribution.
Compiler Engineer - MLIR / PyTorch Infrastructure

Compiler Engineer - MLIR / PyTorch Infrastructure

Mythic

Austin, TX

$106K - $139K/yr

Full-time

Posted 28 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