Software Engineer II, Machine Learning

About the Role: 
We are seeking a passionate and skilled Machine Learning Engineer to work on developing and improving our AI-powered solutions. This position will involve training production models for a variety of machine learning tasks such as forecasting, anomaly detection, and event prediction, as well as leveraging cutting-edge libraries and technologies to build efficient and scalable systems. 
As a Software Engineer II - Machine Learning, you will:

• Train and deploy production-level machine learning models focused on: 
  • Forecasting 
  • Anomaly Detection 
  • Event Prediction 
• Develop and implement machine learning algorithms, ensuring their scalability, performance, and robustness. 
• Create agentic language model based user experiences. 
• Work with large datasets, process and analyze data using tools like Pandas and Numpy. 
• Use modern deep learning frameworks such as Pytorch to implement and optimize models. 
• Integrate machine learning models into the company’s Time and Attendance and Employee Scheduling software. 
• Collaborate with cross-functional teams to solve complex business challenges through AI/ML solutions. 
• Optimize and fine-tune models for performance in real-world production environments. 

You are a strong fit for this role if you have: 

• 2+ years of experience in training and deploying production models 
• 2+ years of experience with the following libraries: 
   
  
  • Pytorch 
  • Pandas 
  • Numpy 
    
• Hands-on experience with language models 
• Solid understanding of the following model architectures: 
  • Transformers 
  • RNN (Recurrent Neural Networks) 
  • S4 | Mamba 
  • LSTM (Long Short-Term Memory Networks) 

Strongly Encouraged: 

• Familiarity with advanced machine learning concepts, including: 
  • State Space Modeling (S4, Mamba) 
  • Hidden Markov Models (HMMs) 
  • BEAM Search 
  • Human Feedback Reinforcement Learning (RLHF) 
  • Relationship Graph Theory 
  • Automatic Prompt Optimization 
• Worked on language model user experiences using: 
  • DSPy 
  • LangChain 

Physical Requirements:

• Prolonged periods sitting at a desk and working on a computer.
• Must be able to lift up to 15 pounds at times.

Benefits: 

• Competitive salary 
• PTO and Sick leaves 
• In-Patient Health insurance 
• Out-Patient Department coverage 
• Provident fund and EOBI 
• The work/life setup you need to be successful. 
• Access to an in-house gym to support your health and well-being 
• A creative, collaborative, supportive environment that gives you the autonomy to explore new ideas, grow your skill set and create outstanding results  
• The chance to make a genuine impact on the company’s growth  
• Plenty of challenging work and the opportunity to stretch yourself  
• The opportunity to work with amazing talent in a fast-growing company that really values their team