Machine Learning Application Engineer

Job Description

We are seeking a highly skilled and motivated Machine Learning Engineer proficient in deep learning frameworks such as PyTorch and TensorFlow. The ideal candidate will have experience deploying machine learning solutions and a passion for staying at the forefront of technology by exploring new advancements in the field. Utilise ML frameworks to build and deploy solutions for clients using LLMs and DL networks. The role involves working with cutting edge Large Language Models (LLMs) and Deep Learning (DL) networks to deliver high-quality, innovative solutions that meet the evolving needs of our clientele.

Responsibilities

• Gain a comprehensive understanding of the various architectures related to large langauage models (LLMs) and conduct research to contribute to the field.
• Deploy and test different model architectures, requiring a solid grasp of deep learning frameworks.
• Develop efficient and innovative machine learning solutions tailored to client needs.
• Drive innovation in the LLM market by creating alternative, unique solutions.

Requirements

• Proficiency in at least one deep learning framework: PyTorch or TensorFlow.
• Strong understanding of the mathematical principles underlying these frameworks.
• Knowledge of the architecture based on various LLMs or similar models.
• Basic Linux and Python skills to efficiently implement standard practices in project collaboration.
• Familiarity with distributed processing systems is beneficial.
• Experience with Gradio or Streamlit libraries is advantageous.

Qualifications

• Demonstrated ability to work independently and in a team environment.
• Excellent problem-solving and communication skills.

About shodh.ai and the Job

We are India's only Deep Learning product building lab. We have developed our own pipelline and framework for pre-training our own LLMs. Our LLMs are built from scratch. We have our own H100 clusters and are currently building tools for various unicorns, companies and well as DRDO. The founder team includes ex-Microsoft Research scientists and ex-DRDO scientists.

Place: Remote (Company locations in Jaipur/Bangalore) Eligibility: Open to all branches and CGPA DOJ: At first availability Stipend: Competitive PPO: Based on performance