Full-Time
Machine Learning Engineer
Confirmed live in the last 24 hours
Accelerates AI startups from inception to funding
Venture Capital
Senior
San Diego, CA, USA
Required Skills
Python
Tensorflow
Git
Pytorch
AWS
Pandas
Requirements
- 4+ years programming experience with Python
- Proven experience as an ML Engineer working on deploying production model training and/or inference
- Excellent knowledge of the DL tools PyTorch and TensorFlow
- Excellent knowledge and experience of using DL systems libraries such as FSDP, DDP, DeepSpeed, and GPipe
- Familiarity with LLMs, finetuning, and associated concepts
- Familiarity with operating systems concepts, memory management, networking, and cloud computing
- Familiarity with AWS infrastructure components like EC2, S3, EBS, EFS, EKS, and Lambda
- Basic experience with version control systems (e.g., Git) and collaborative development workflows
- Understanding of CI/CD methodologies and tools
- Excellent communication and collaboration skills
- Ability to work independently and as part of a team
- Strong problem-solving and analytical skills
- A passion for learning and staying updated with the latest ML technologies
Responsibilities
- Design, develop, deploy, and maintain large-scale DL infrastructure software
- Contribute to designing efficient distributed systems that can scale DL computations
- Automate the set up, launch, and orchestration of end-to-end training and experimentation pipelines
- Use and extend libraries like FSDP, DDP, DeepSpeed, and GPipe to train DL models across multiple GPUs
- Use tools like Pandas and Dask to handle large multimodal datasets
- Troubleshoot code and fix bugs to ensure smooth functioning of the application
- Monitor and troubleshoot cluster resource usage to ensure optimal performance
- Conform to continuous integration and continuous delivery (CI/CD) pipeline standards for code deployment
- Communicate effectively with the wider team to ensure successful application development and deployment
- Collaborate with other developers to define and implement cloud infrastructure strategies for DL applications
- Stay up-to-date with the latest advancements in DL and AI technologies and best practices
AI Fund collaborates with entrepreneurs to accelerate startups from idea to Series A funding by leveraging artificial intelligence and machine learning technologies. The company empowers innovators to solve challenging problems with AI, focusing on the critical and risky phase from 0 to 1.
Company Stage
N/A
Total Funding
N/A
Headquarters
East Palo Alto, California
Founded
2017
Growth & Insights
Headcount