CrawlJobs Logo

Software Engineer, Infra PyTorch (PhD)

United States, Menlo Park 181000.00 USD / Year · Job Posted February 04, 2026
Apply Position
Job Link Share

Job Description

This role is about developing the core PyTorch 2.0 technologies, innovating and advancing the state-of-the-art of ML compilers, and accelerating PT2 adoption through direct engagements with OSS and industry users.The PyTorch Compiler team is dedicated to making PyTorch run faster and more resource-efficient without sacrificing its flexibility and ease of use. The team is the driving force behind PT2, a step function change in PyTorch’s history that brought compiler technologies to the core of PyTorch. PT2 technologies have gained industry-wide recognition since their first release in March 2023. The team is committed to building the PT2 compiler that withstands the test of time while striving to become the #1 ML framework compiler in the industry. Our work is open source, cutting-edge, and industry leading.

Job Responsibility

  • Develop the PT2 compiler (e.g., TorchDynamo, TorchInductor, PyTorch Distributed, PyTorch Core)
  • Improve PyTorch performance via systematic solutions for the entire community
  • Explore the intersection of the PyTorch compiler and PyTorch distributed
  • Optimize Generative AI models across the stack (pre-training, fine-tuning, and inference)
  • Collaborate with users of PyTorch to enable new use cases of PT2 technologies both inside and outside Meta

Requirements

  • Currently has, or is in the process of obtaining a Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience. Degree must be completed prior to joining Meta
  • Currently has or is in the process of obtaining a PhD degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience. Degree must be completed prior to joining Meta
  • Research or industry experience in developing compilers, ML systems, ML accelerators, GPU performance, and similar
  • Advanced in Python or C++ programming

Nice to have

  • Experience in developing PyTorch/PT2, Triton, MLIR, JAX, XLA, TVM is a huge plus
  • Knowledge in GPU architecture, ML accelerator performance, and developing high-performance kernels
  • Experience in building OSS communities and extensive social media presence in the ML Sys domain
  • Experience with training models, end-to-end model optimizations, or applying ML to systems
  • Knowledge of communication collectives, PyTorch distributed, and parallelism
  • Experience in developing inside other ML frameworks like Caffe2, TensorFlow, ONNX, TensorRT
  • First-authored publications at peer-reviewed conferences (e.g. NeurIPS, MLSys, ASPLOS, PLDI, ICML, or similar)

What we offer

  • bonus
  • equity
  • benefits

Looking for more opportunities?

Search for other job offers that match your skills and interests.

Similar Jobs for

Software Engineer, Infra PyTorch (PhD)

8 matching positions

Principal Engineer - Marketplace

Principal Engineer role in the Marketplace Engineering team to lead breakthrough...
Location
Location
United States , San Francisco; Sunnyvale
Salary
Salary:
302000.00 - 336000.00 USD / Year
uber.com Logo
Uber
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • PhD in Computer Science, Machine Learning, Operations Research, or related quantitative field OR Master’s degree with 12+ years of industry experience
  • 10+ years of experience building and deploying ML models in large-scale production environments
  • Expert-level proficiency in modern ML frameworks (TensorFlow, PyTorch, JAX) and distributed computing platforms (Spark, Ray)
  • Deep expertise across multiple areas including: Deep Learning, Causal Inference, Reinforcement Learning, Multi-objective Optimization, Algorithmic Game Theory, and Large-scale Ads Ranking/Auction Systems
  • Proven track record of leading complex ML projects from research through production with significant measurable business impact
  • Strong programming skills in Python, Java, or Go with experience building production ML systems
  • Experience with feature engineering, model serving, and ML infrastructure at scale (handling millions of predictions per second)
  • Technical leadership experience including mentoring senior engineers and driving cross-team technical initiatives
  • Advanced Deep Learning and Neural Network architectures
  • Scalable ML architecture and distributed model training
Job Responsibility
Job Responsibility
  • Lead the design and implementation of advanced ML systems for dynamic pricing algorithms serving millions of drivers across 70+ countries around the world
  • Architect real-time ML infrastructure handling 1M+ pricing decisions per second with sub-50ms latency requirements
  • Drive breakthrough research in causal ML, reinforcement learning, algorithmic game theory, and multi-objective optimization for marketplace optimization with strategic agents
  • Own end-to-end ML model lifecycle from research through production deployment and continuous optimization
  • Develop and enforce best practices in system design, ensuring data integrity, security, and optimal performance
  • Serve as a representative for the Marketplace organization to the broader internal and external technical community
  • Contribute to the eng brand for Marketplace and serve as a talent magnet to help attract and retain talent for the team
  • Stay abreast of industry trends and emerging technologies in software engineering, focused particularly on ML/AI, to enhance our systems and processes continually
  • Build scalable ML architecture and feature management systems supporting Driver Pricing and broader Marketplace teams
  • Design experimentation frameworks enabling rapid testing of pricing algorithms using A/B, Switchback, Synthetic Control, and other experimental methodologies
What we offer
What we offer
  • Eligible to participate in Uber's bonus program
  • May be offered an equity award & other types of comp
  • Eligible to participate in a 401(k) plan
  • Eligible for various benefits (details at provided link)
  • Fulltime
Read More
Arrow Right
New

Pharmacy Technician

We're building a world of health around every individual — shaping a more connec...
Location
Location
United States , Collierville
Salary
Salary:
17.00 - 25.65 USD / Hour
https://www.cvshealth.com/ Logo
CVS Health
Expiration Date
July 31, 2026
Flip Icon
Requirements
Requirements
  • Valid Tennessee Pharmacy Technician Registration
Job Responsibility
Job Responsibility
  • Assists customers by responding to specialty product inquiries and providing technical advice or recommendations for specialty needs
  • Maintains specialty retail store operations by completing inventory management tasks, restocking shelves, and organizing displays
  • Communicates with customers and shares announcements and/or promotions on the latest product releases, features, and technology trends
  • Compiles customer complaints, inquiries, or concerns for senior managers to help improve future customer service outcomes
  • Monitors key sales metrics related to daily, weekly, and monthly sales revenue, units sold, average transaction value, and conversion rates
  • Operates within the point-of-sale (POS) system to process transactions, handle cash, and operate payment terminals
  • Drives effective team collaborations with specialty retail associates and technicians to provide customers with a positive experience
  • Determines how best to prioritize tasks and manage workloads to ensure the timely completion of customer requests and specialty retail services
  • Performs basic repairs or provides step-by-step instructions to customers to help resolve issues with specialty retail products and/or services
What we offer
What we offer
  • Medical, dental, and vision coverage
  • paid time off
  • retirement savings options
  • wellness programs
  • 1500 USD bonus
  • Fulltime
Read More
Arrow Right
New

Electrical Technician II

Amentum is seeking an Electrical Technician II to join our operations & maintena...
Location
Location
United States , New Albany
Salary
Salary:
30.00 - 32.00 USD / Hour
amentum.com Logo
Amentum
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • High school diploma or demonstrated equivalent
  • 3 - 4 years’ experience as an electrician in a data center, industrial environment, hospital, manufacturing facility or demonstrated equivalent combination of education and experience
  • Must have the ability to work in an aerial lift or scissor lift at heights up to 30 feet or higher
  • Must be self-motivated and demonstrate the strong organizational/time management skills needed to be successful in this role
  • Qualified applicants who are offered a position must pass a pre-employment background check and substance abuse test
  • This position will require the ability to obtain a security access badge at our client’s location
  • This position may require a valid Driver’s license
  • This position may require various shifts, weekends, and alternative work schedules
Job Responsibility
Job Responsibility
  • Maintain current and future facility electrical systems
  • Diagnoses, troubleshoots, maintains, and repairs electrical equipment and control circuits from 24v up to and including 480V AC, electro-mechanical test devices, instrumentation devices, PLCs / ladder logic controllers, and variable frequency drives as necessary
  • Comfortable working near or around 13.8K voltage
  • Support facility electrical expansion projects
  • Inspects, maintains, and repairs various critical infrastructure systems such as servers, network equipment, Programmable Logic Controllers (PLCs), system administration, access control, alarm investigation, and SCADA management
  • Provide daily assistance to technicians as you read blueprints/schematics/ladder logic/sequence of operations, conduct tours of systems, and assess their working order
  • Must be able to adhere to Federal, State and Local electrical code and be familiar with NFPA70e
  • Demonstrates ability to identify and correct any safety-related issues and perform independent equipment evaluations to identify potential equipment failures
  • Performs preventive maintenance on process/production equipment as scheduled in the Computerized Maintenance Management System (CMMS)
  • Responds and provides service and feedback to the customer on all work orders while assuring compliance to codes, regulations, and industry standards
What we offer
What we offer
  • Health, dental, and vision insurance
  • Paid time off and holidays
  • Retirement benefits (including 401(k) matching)
  • Educational reimbursement
  • Parental leave
  • Employee stock purchase plan
  • Tax-saving options
  • Disability and life insurance
  • Pet insurance
  • Fulltime
Read More
Arrow Right
New

Pharmacist - Specialty Mail Order

We’re building a world of health around every individual — shaping a more connec...
Location
Location
United States , Bartlett
Salary
Salary:
45.43 - 76.15 USD / Hour
https://www.cvshealth.com/ Logo
CVS Health
Expiration Date
June 30, 2026
Flip Icon
Requirements
Requirements
  • Pharmacy degree from an accredited college of pharmacy
  • Active license as a practicing pharmacist in TN
  • 1+ years working experience as a licensed Pharmacist
Job Responsibility
Job Responsibility
  • Provide exceptional customer service to patients and healthcare professionals
  • Provide leadership and guidance to supporting staff
  • Interact with health care professionals in the provision and coordination of customer care
  • Direct supportive personnel in daily activities pertaining to clinical and distributive activities
  • Ensure activities are performed to CVS Health standards and in accordance with written SOP’s and regulatory constraints
  • Verify orders prepared for delivery to prevent errors
  • Obtain nonresident licenses to meet business needs
What we offer
What we offer
  • Vacation
  • Paid holidays
  • Health/dental/vision insurance
  • Employee discounted stock purchase program
  • 401K
  • Tuition reimbursement
  • 20-30% employee discount at all CVS stores
  • Bonus, commission or short-term incentive program
  • Medical, dental, and vision coverage
  • Paid time off
  • Fulltime
Read More
Arrow Right
New

Teaching Assistant

As a Primary Teaching Assistant, you will play a key role in supporting the lear...
Location
Location
United Kingdom , Redhill
Salary
Salary:
90.00 - 95.00 GBP / Day
https://www.randstad.com Logo
Randstad
Expiration Date
June 06, 2026
Flip Icon
Requirements
Requirements
  • Experience working with children, preferably in a primary school setting
  • A patient, caring, and supportive approach to working with young learners
  • Strong communication and teamwork skills
  • Flexibility, adaptability, and a positive attitude
  • Relevant qualifications in education or childcare are advantageous but not essential
  • A commitment to safeguarding and promoting the welfare of children (enhanced DBS required)
Job Responsibility
Job Responsibility
  • Work closely with class teachers to deliver engaging lessons
  • Provide one-on-one or small group support to students needing additional help
  • Assist with classroom management and help maintain a positive learning environment
  • Support children with special educational needs (SEN) when required
  • Prepare and organise teaching materials and resources
  • Contribute to school activities, events, and parent meetings as needed
What we offer
What we offer
  • Competitive Pay
  • Flexible Hours
  • Professional Development
  • Supportive Environment
  • Variety of Roles
  • training and education
!
Read More
Arrow Right
New

Primary Teaching Assistant

Primary Teaching Assistant - Randstad Education, Redhill. Location: Redhill and ...
Location
Location
United Kingdom , Redhill
Salary
Salary:
90.00 - 95.00 GBP / Day
https://www.randstad.com Logo
Randstad
Expiration Date
June 28, 2026
Flip Icon
Requirements
Requirements
  • Previous background in a childcare or primary school environment
  • A dedicated focus on child protection and the safety of all students
  • Formal education or childcare certifications are preferred but not required
Job Responsibility
Job Responsibility
  • Collaborating with teaching staff to execute lessons and manage classroom logistics
  • Delivering tailored academic support to individuals or small clusters of learners
  • Implementing effective behavioral strategies to preserve a calm and focused classroom
  • Assisting pupils with Special Educational Needs (SEN) to ensure curriculum access
  • Fostering a welcoming, diverse, and encouraging school culture
  • Engaging in staff briefings, school functions, and planning sessions
What we offer
What we offer
  • Performance-Based Pay
  • Schedule Flexibility
  • Career Growth
  • Team Dynamics
  • Local Network
  • Training and education
Read More
Arrow Right
New

Mathematics Teacher

Maths Teacher | Long-Term & Temporary Opportunities | Redhill. Are you a passion...
Location
Location
United Kingdom , Redhill
Salary
Salary:
162.00 - 252.00 GBP / Day
https://www.randstad.com Logo
Randstad
Expiration Date
July 08, 2026
Flip Icon
Requirements
Requirements
  • Qualified Teacher Status (QTS) or equivalent (ECTs/NQTs are highly encouraged to apply!)
  • degree in Mathematics or a related subject
  • a proven track record of helping students achieve strong results
  • an Enhanced DBS certificate (on the Update Service is a massive plus)
  • a resilient, patient, and professional attitude
Job Responsibility
Job Responsibility
  • Plan and deliver engaging, student-centered lessons that cater to a range of abilities
  • Use data and assessments to track student growth and provide 'live' feedback to close learning gaps
  • Maintain a positive, 'outstanding' learning environment where students feel safe and focused
  • Work closely with department heads and Teaching Assistants to ensure every child reaches their potential
  • Act as a vigilant protector of student well-being, following all school policies and procedures
What we offer
What we offer
  • Referral Bonus
  • flexible schedule
  • dedicated consultant with local knowledge of Redhill schools
  • access to free CPD training, including Safeguarding and Team Teach
  • earn £300 in Love2Shop vouchers for every teacher you refer
  • Fulltime
Read More
Arrow Right
New

Teacher Of English

Randstad Education is looking for an enthusiastic and innovative English Teacher...
Location
Location
United Kingdom , Redhill
Salary
Salary:
160.00 - 252.00 GBP / Day
https://www.randstad.com Logo
Randstad
Expiration Date
July 04, 2026
Flip Icon
Requirements
Requirements
  • Qualified Teacher Status (QTS) or equivalent
  • A deep passion for English Language and Literature with the ability to lead a room with presence and authority
  • Excellent written and spoken English, acting as a role model for oracy and language use
  • An Enhanced DBS check is required (Randstad can assist with your application)
  • An honest, approachable educator who prioritizes child safety and follows all safeguarding and reporting protocols
Job Responsibility
Job Responsibility
  • Lead engaging English lessons across Key Stages 3 & 4, making complex texts accessible and fun for all learners
  • Focus on extending language and increasing vocabulary, using visual materials and sentence stems to support student progress
  • Identify students who may be falling behind and provide the 'extra support' needed to build their confidence and academic success
  • Maintain a non-judgmental, encouraging classroom where students feel secure to make mistakes and learn at their own pace
  • Work as a strong team player, sharing best practices and contributing to the school's overall 'Working Together' ethos
What we offer
What we offer
  • Competitive Daily Rates / MPS / UPS
  • PAYE
  • Referral Bonus
  • Training and education
  • Safeguarding & Prevent
  • Fulltime
Read More
Arrow Right