ZipRecruiter

Log In

1

Pytorch Developer Jobs in Austin, TX (NOW HIRING)

Targeted Talent

Senior Deep Learning Engineer

Austin, TX · On-site +1

$130K - $180K/yr

We're seeking top-notch engineers to join our team. As part of our group, you'll collaborate with ... Proficiency in deep learning frameworks like Tensorflow and/or PyTorch * Experience with CNNs ...

Kaav Inc.

Java AI Engineer-C2C

Austin, TX · On-site

$51.25 - $70.25/hr

Java AI Engineer-NO OPT and H1B Location: Austin, TX/ Sunnyvale, CA (Sunnyvale - 2 Positions ... Familiarity with AI frameworks and libraries (e.g., TensorFlow, PyTorch, OpenCV)

Kaav Inc

Java AI Engineer-C2C

Austin, TX · On-site

$51.25 - $70.50/hr

Java AI Engineer-NO OPT and H1B Location: Austin, TX/ Sunnyvale, CA (Sunnyvale - 2 Positions ... Familiarity with AI frameworks and libraries (e.g., TensorFlow, PyTorch, OpenCV)

Gainwell Technologies LLC

Expert AI Engineer

Austin, TX · On-site

$147K - $210K/yr

Strong programming skills in Python, TensorFlow, PyTorch, and other AI frameworks. * Strong problem-solving skills with the ability to translate business challenges into AI-driven solutions. What we ...

Allen Control Systems

Software Engineer

Austin, TX · On-site

... PyTorch) or scientific computing. • Experience with DevOps tools (Docker, Kubernetes, CI/CD pipelines) Company : Allen Control Systems develops autonomous defense technologies designed to detect ...

Allen Control Systems

Software Engineer

Austin, TX · On-site

... PyTorch) or scientific computing. • Experience with DevOps tools (Docker, Kubernetes, CI/CD pipelines) Company : Allen Control Systems develops autonomous defense technologies designed to detect ...

Nvidia

Senior ML Compiler Engineer

Austin, TX

$103K - $142K/yr

PyTorch, JAX etc) * Strong python and C/C++ programming skills Ways to stand out from the crowd: * Expertise in AI frameworks such as PyTorch, TensorFlow, and ONNX * Expertise in machine learning ...

New

Vkore Solutions

GenAI Engineer-W2

Austin, TX · On-site

We are looking for a GenAI Ops Engineer to train, fine-tune, and deploy Generative AI models (LLMs ... Train and fine-tune LLMs using PyTorch, DeepSpeed, and LoRA. * Optimize inference using ONNX, vLLM ...

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.
Software Engineer II: AI Compiler Engineer

Software Engineer II: AI Compiler Engineer

Cadence Design Systems Inc.

Austin, TX • Hybrid

$96K - $132K/yr

Full-time

Posted 12 days ago

Job description

At Cadence, we hire and develop leaders and innovators who want to make an impact on the world of technology.

Job Description

Cadence Design Systems Inc. is looking for a motivated Software Engineer II: AI Compiler Engineer to work with us.

As a Software Engineer II: AI Compiler Engineer you will work with complex high performance SoC's, and is one of the best kept secrets within the semi IP world powering AR/VR, HiFi Audio and Speech, Vision, Imaging and hundreds of intelligent IoT applications.

Be a part of a team that develops an AI graph compiler that takes as input Neural Networks (NNs) created in frameworks such as PyTorch or TensorFlow and converts them into optimized code suitable for execution on special-purpose and embedded platforms.

Cadence is also a Fortune 100 Best Companies to Work For.

Job Description:

  • Developing a deep learning compiler stack that takes neural network descriptions (CNNs/RNNs) created in frameworks such as Caffe, PyTorch, TensorFlow, etc. and converts them into code suitable for execution on special-purpose and embedded platforms.
  • Use modern compiler frameworks such as LLVM and MLIR.
  • Developing optimized implementations of a variety of neural-network operations and integrating them into a runtime framework
  • Developing new optimization techniques and algorithms to efficiently map CNNs onto a wide range of Xtensa processors and specialized hardware.
  • Benchmarking end-to-end network performance on a variety of DSP and special-purpose accelerator platforms.
  • Enhancing the framework to improve overall functionality and performance on the various hardware platforms.
  • Devising multiprocessor/multicore partitioning and scheduling strategies.
  • Developing complex programs to validate the functionality and performance of the CNN application programming kit.
  • Working with hardware designers to identify opportunities for additional hardware acceleration of neural network functions.
  • Working with industry-leading partners and customers to design and standardize neural network APIs..

Requirements:

  • Complete Bachelor in Computer Science or Computer Engineering or equivalent experience.
  • A high level of C and C++ programming expertise with 3-5+ years of experience is required.
  • Expertise in software development on Linux and Windows systems including test, debug and release is required.
  • Knowledge of and experience with a state-of-the-art compiler stack such as LLVM and MLIR.
  • Experience implementing compilation techniques such loop optimization, polyhedral models, and IR construction/transition/lowering techniques.

Nice to have:

  • Master or PhD.
  • 3+ years of experience working on a production compiler is highly desired.
  • Python experience highly desired
  • Prior work with CNNs and familiarity with deep learning frameworks (TensorFlow, Caffe/2, etc.) is a strong plus
  • Experience programming and optimizing for embedded platforms such as DSPs with DMA engines highly desired
  • Familiarity with the state-of-the-art deep learning compilation approaches (Glow, TVM, XLA, etc.) is a plus
  • Familiarity with various deep learning networks and their applications (Classification/Segmentation/Object Detection/RNNs) is a plus
  • Knowledge of neural net exchange formats (ONNX, NNEF) is a plus

Additional Job Details:

  • Employment term: 40 hours/week.
  • Hybrid work.
  • Competitive benefits.

Cadence is the only company that provides the expertise and tools, IP, and hardware required for the entire electronics design chain, from chip design to chip packaging to boards and to systems. We enable electronic systems and semiconductor companies to create innovative products that transform the way people live, work, and play. Our products are used in mobile, consumer, cloud datacenter, automotive, aerospace, IoT, industrial and other market segments.

For more information, access http://www.cadence.com

We're doing work that matters. Help us solve what others can't.

Apply