Austin, TX · On-site
$133K - $179K/yr
Establish profiling models that integrate with existing ML/AI workflows (e.g., PyTorch/XLA) to turn low-level signals into actionable insights. * Set technical direction for an engineering team ...
Austin, TX · On-site
$133K - $179K/yr
Establish profiling models that integrate with existing ML/AI workflows (e.g., PyTorch/XLA) to turn low-level signals into actionable insights. * Set technical direction for an engineering team ...
Experience with frameworks like PyTorch or TensorFlow is a plus. * Infrastructure/DevOps: Exposure to cloud platforms (AWS/Azure/GCP) and containerization (Docker/Kubernetes). * Analytical Mindset:
Experience with frameworks like PyTorch or TensorFlow is a plus. * Infrastructure/DevOps: Exposure to cloud platforms (AWS/Azure/GCP) and containerization (Docker/Kubernetes). * Analytical Mindset:
Austin, TX · On-site
$90 - $130/hr
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 ...
Austin, TX · On-site
$90 - $130/hr
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 ...
Austin, TX · On-site
$96K - $132K/yr
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 ...
Austin, TX · On-site
$96K - $132K/yr
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 ...
Austin, TX · On-site
$96K - $132K/yr
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 ...
Austin, TX · On-site
$96K - $132K/yr
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 ...
Austin, TX · On-site
... learn, PyTorch, TensorFlow) - Experience with data engineering fundamentals (ETL/ELT, SQL, data preprocessing, feature engineering) - Knowledge of model evaluation, validation strategies, and ...
Quick apply
Austin, TX · On-site
... learn, PyTorch, TensorFlow) - Experience with data engineering fundamentals (ETL/ELT, SQL, data preprocessing, feature engineering) - Knowledge of model evaluation, validation strategies, and ...
Austin, TX · On-site
Hands-on experience with profiling tools such as NVIDIA Nsight, PyTorch Profiler, or Intel Gaudi instrumentation.
Quick apply
Austin, TX · On-site
Hands-on experience with profiling tools such as NVIDIA Nsight, PyTorch Profiler, or Intel Gaudi instrumentation.
Austin, TX · Hybrid
$96K - $132K/yr
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 ...
Austin, TX · Hybrid
$96K - $132K/yr
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 ...
Austin, TX · Hybrid
$96K - $132K/yr
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 ...
Austin, TX · Hybrid
$96K - $132K/yr
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 ...
Austin, TX · On-site
$96K - $132K/yr
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 ...
Austin, TX · On-site
$96K - $132K/yr
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 ...
Austin, TX · On-site +1
$130K - $180K/yr
Proficiency in deep learning frameworks like Tensorflow and/or PyTorch * Experience with CNNs, LSTMs/RNNs, Transformers * Strong math skills and Python proficiency * Experience with C/C++ Preferred ...
Quick apply
Austin, TX · On-site +1
$130K - $180K/yr
Proficiency in deep learning frameworks like Tensorflow and/or PyTorch * Experience with CNNs, LSTMs/RNNs, Transformers * Strong math skills and Python proficiency * Experience with C/C++ Preferred ...
Austin, TX · On-site
$103K - $142K/yr
Deep proficiency in PyTorch, TensorFlow, or Hugging Face Transformers. * Proven experience deploying AI models across cloud, edge, and mobile hardware environments. * Expertise in model compression ...
Austin, TX · On-site
$103K - $142K/yr
Deep proficiency in PyTorch, TensorFlow, or Hugging Face Transformers. * Proven experience deploying AI models across cloud, edge, and mobile hardware environments. * Expertise in model compression ...
Austin, TX · On-site
$103K - $142K/yr
Deep proficiency in PyTorch, TensorFlow, or Hugging Face Transformers. * Proven experience deploying AI models across cloud, edge, and mobile hardware environments. * Expertise in model compression ...
Austin, TX · On-site
$103K - $142K/yr
Deep proficiency in PyTorch, TensorFlow, or Hugging Face Transformers. * Proven experience deploying AI models across cloud, edge, and mobile hardware environments. * Expertise in model compression ...
Round Rock, TX · On-site
... PyTorch, Scikit-learn, or similar ML frameworks. • Strong understanding of NLP, Generative AI, LLMs, and/or computer vision. • Experience developing and deploying production-grade ML models. • ...
Round Rock, TX · On-site
... PyTorch, Scikit-learn, or similar ML frameworks. • Strong understanding of NLP, Generative AI, LLMs, and/or computer vision. • Experience developing and deploying production-grade ML models. • ...
Austin, TX · On-site
$140 - $200/hr
Deep proficiency in PyTorch, TensorFlow, or Hugging Face Transformers. * Proven experience deploying AI models across cloud, edge, and mobile hardware environments. * Expertise in model compression ...
Austin, TX · On-site
$140 - $200/hr
Deep proficiency in PyTorch, TensorFlow, or Hugging Face Transformers. * Proven experience deploying AI models across cloud, edge, and mobile hardware environments. * Expertise in model compression ...
Austin, TX · On-site
$185K - $335K/yr
Deep expertise in PyTorch 2.x+ and distributed training frameworks. * Experience designing and developing training platforms that support FSDP, pipeline parallelism, and other scalable solutions for ...
Austin, TX · On-site
$185K - $335K/yr
Deep expertise in PyTorch 2.x+ and distributed training frameworks. * Experience designing and developing training platforms that support FSDP, pipeline parallelism, and other scalable solutions for ...
Austin, TX · On-site
$120 - $160/hr
Optimize AI and compute workloads for AMD GPUs and APUs (e.g., ROCm, ONNX, PyTorch, inference runtimes) * Apply a data‑driven approach to profiling, benchmarking, and tuning * Identify gaps in the ...
Austin, TX · On-site
$120 - $160/hr
Optimize AI and compute workloads for AMD GPUs and APUs (e.g., ROCm, ONNX, PyTorch, inference runtimes) * Apply a data‑driven approach to profiling, benchmarking, and tuning * Identify gaps in the ...
Austin, TX · On-site
Must-Have Skills 3+ years of ML engineering experience -- model training, fine-tuning, or post-training pipelines in research or production Strong Python and deep learning proficiency (PyTorch ...
Austin, TX · On-site
Must-Have Skills 3+ years of ML engineering experience -- model training, fine-tuning, or post-training pipelines in research or production Strong Python and deep learning proficiency (PyTorch ...
San Marcos, TX · On-site
Must-Have Skills 3+ years of ML engineering experience -- model training, fine-tuning, or post-training pipelines in research or production Strong Python and deep learning proficiency (PyTorch ...
San Marcos, TX · On-site
Must-Have Skills 3+ years of ML engineering experience -- model training, fine-tuning, or post-training pipelines in research or production Strong Python and deep learning proficiency (PyTorch ...
Austin, TX · On-site
Required : • 3-6 years of experience in applied ML, with at least 1-2 years working with LLMs • Strong Python skills and familiarity with ML/LLM frameworks (PyTorch, Transformers, LangChain ...
Austin, TX · On-site
Required : • 3-6 years of experience in applied ML, with at least 1-2 years working with LLMs • Strong Python skills and familiarity with ML/LLM frameworks (PyTorch, Transformers, LangChain ...
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.
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.
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.
For Pytorch jobs in Austin, TX, the most frequently searched job titles are:
The top searched job categories for Pytorch jobs in Austin, TX are:
Cities near Austin, TX with the most Pytorch job openings:
$133K - $179K/yr
Full-time
Re-posted 9 days ago
9.6
Based on 17 frontline employees who took The Breakroom Quiz
7th of 245 rated software companies
Help design and ship an Always-On, low-overhead GPU profiling service that runs in production, scales across cluster environments, and delivers actionable insights for ML workloads. You will lead the architecture and hands-on delivery across system software, drivers, and CUDA to make profiling continuously available and reliable.
What you'll be doing:
Design the architecture for an Always-On profiling service, defining interfaces, data flows, and scalability guarantees for multi-process/GPU/node systems.
Drive low-overhead, high-reliability implementations in C/C++, including IPC/shared memory, and bounded CPU/memory budgets.
Lead end-to-end feature delivery spanning user-mode components, driver/platform layers, and performance counter/trace providers.
Establish profiling models that integrate with existing ML/AI workflows (e.g., PyTorch/XLA) to turn low-level signals into actionable insights.
Set technical direction for an engineering team; mentor engineers, drive technical planning to mitigate architectural risks, and align roadmaps across internal and external partners.
What we need to see:
BS or MS degree or equivalent experience in Computer Engineering, Computer Science, or related degree.
15+ years of system-level C/C++ development, including concurrency, memory management, and performance engineering.
Proven experience designing and shipping production quality system software or drivers with strict reliability, observability, and performance constraints.
Demonstrated technical leadership: defining architecture and success metrics, and translating abstract product visions into actionable technical roadmaps with fast-paced, multidisciplinary teams.
Strong interpersonal, verbal, and written communication; able to influence across organizations and build trust with external collaborators.
Ways to stand out from the crowd:
Extensive experience with profiling/tracing stacks for CPU/GPU (e.g., CUPTI, Nsight, performance counters, event correlation) and debugging highly concurrent systems.
Deep hands-on knowledge of CUDA and GPU architecture, including runtime/driver APIs, CUDA streams/graphs, and kernel behavior.
Track record building continuous, always-on, or multi-client profiling systems designed for predictable overhead at scale.
Hands-on experience tuning ML training/inference loops based on deep profiling analysis, with familiarity in ML ecosystems (e.g., PyTorch, JAX) and correlating application events with GPU metrics to translate data into actionable performance insights (e.g., bottleneck triage, compute vs. memory bound).
Experience with user-mode driver development and integration within platform security and permissions models.
You will also be eligible for equity and benefits.
This posting is for an existing vacancy.
NVIDIA uses AI tools in its recruiting processes.
NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.
Sourced by ZipRecruiter
NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology--and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent.
Computer and electronic product manufacturing
10,000+ Employees
Santa Clara, CA, US
1993
