Overview
Company name: Wald.ai | HQ Location: Palo Alto, California | Website | LinkedIn
Role: ML Engineer
- Experience: 3+ years
- Salary: Rs. 25-55 lakhs per annum
- Location: Remote
- Type: Full-time
About the Role
We are looking for a Founding ML Engineer with 3+ years of experience in software development and a strong focus on machine learning, LLMs, and generative AI. This is a unique opportunity to be part of an early-stage AI startup, driving cutting-edge AI innovations and shaping the future of our ML infrastructure and products.
Key Responsibilities
- Develop and deploy ML models, focusing on LLMs, generative AI, and model fine-tuning.
- Build and optimize machine learning pipelines, ensuring scalability and efficiency.
- Work with large-scale datasets, improving data processing, feature engineering, and training workflows.
- Collaborate with cross-functional teams to integrate AI solutions into production systems.
- Stay ahead of AI/ML advancements, researching and implementing the latest techniques.
- Optimize model inference performance for real-world applications.
- Ensure robust monitoring, evaluation, and continuous improvement of deployed models.
What We’re Looking For
- 3+ years of hands-on experience in software development with a strong focus on machine learning.
- Proficiency in Python and deep understanding of ML frameworks (TensorFlow, PyTorch, Hugging Face, etc.).
- Experience with LLMs, generative AI, and model fine-tuning is highly desirable.
- Familiarity with cloud platforms (AWS, GCP, or Azure) and ML infrastructure.
- Understanding of MLOps best practices, including model versioning, monitoring, and deployment.
- Strong problem-solving skills and ability to work in a fast-paced startup environment.