AI ML Developer

Job ID: TM1

Onsite/Local (PA/CA) AI/ML Developer with Python, PyTorch, GPU coding, CI/CD, GitHub,big deep learning models training, code optimization experience

Location: Pittsburgh, PA or Burlingame, CA
Duration: 12 Months

Required Skills:

6+ years of experience in Python, especially with large codebases, preferably in Big Tech or mid-to-large companies
Experience in training big deep learning models with PyTorch
Proficiency in continuous integration and testing
Knowledge of code optimization techniques
Familiarity with GPU coding and backend ML systems (a plus)
Strong problem-solving skills and attention to detail, with the ability to independently overcome technical roadblocks
Experience working collaboratively and communicate effectively across functional teams
Capability to work independently and within a team
Preferred Skills:
Experience in maturing machine learning systems, such as defining evaluation metrics, building evaluation systems, scaling up pipelines and tooling
Experience in creating and maintaining public projects (e.g., GitHub)
Code Optimization (CPU and GPU), GPU Coding