Python ML Engineer - Activeloop

At Activeloop we are transforming the way organizations harness their data for AI with our Deep Lake and Multi-modal AI Search. Whether you're answering critical clinical questions or searching across vast repositories of scientific papers, we empower you to index, search and organize billions of documents, images, and videos intuitively using natural language powered by Large Language Models. Join us in making data more accessible and actionable than ever before.

About the Role

We are looking for a Python ML Engineer with a strong foundation in machine learning, large-scale data systems, and deep learning. The ideal candidate will have expertise in developing and optimizing ML pipelines, implementing efficient indexing techniques, and integrating state-of-the-art retrieval and organization methods. You will collaborate with software engineers, customers, and business stakeholders to develop ML solutions that deliver significant value to the organization and our clients.

Key Responsibilities

Machine Learning Pipeline Development: Design, implement, and optimize robust machine learning pipelines for large-scale datasets.

Algorithm Optimization: Develop and refine algorithms for semantic understanding, retrieval performance, and relevance ranking.

Data Integration: Work on integrating data storage and retrieval solutions within Deep Lake to support efficient data access for ML models.

Query Understanding and Processing: Build advanced pipelines for query processing, including contextual interpretation and intent recognition, to improve data interaction.

Model Development: Fine-tune and deploy machine learning models tailored to data organization and retrieval tasks.

Performance Evaluation: Establish metrics and testing frameworks to continuously evaluate and improve system performance.

Scalability and Efficiency: Optimize ML systems for high throughput, low latency, and large-scale dataset handling.

What We Need to See

• A Master’s or PhD degree in Computer Science, Machine Learning, Statistics, or a related field.

• Strong programming skills in Python and experience with ML libraries such as TensorFlow or PyTorch.

• Proven experience in deploying machine learning models in production environments.

• Knowledge of advanced machine learning algorithms, including deep learning, reinforcement learning, and ensemble methods.

• Solid understanding of data pre-processing, feature engineering, and data quality assurance.

• Excellent problem-solving skills and the ability to extract meaningful insights from complex datasets.

Ways to Stand Out from The Crowd

• Familiarity with Retrieval-Augmented Generation (RAG) techniques.

• Experience with distributed training of deep learning models.

• Publications in top-tier ML and AI conferences like ICML, NeurIPS, or CVPR.

• You’re a highly motivated and curious individual with a builder mindset.

• You thrive in fast-paced environments and enjoy working on meaningful, impactful projects.

• You have a passion for building scalable, impactful ML tools in a startup environment.

• You’re excited to contribute to the startup journey of building an enduring, scalable business.

Why Join Activeloop?

Activeloop Deep Lake is at the forefront of transitioning from traditional software to AI, accelerating AI deployment across various industries. Our products empower advanced LLMs, generative models, and computer vision models. Trusted by industry leaders we are expanding our team to further advance AI applications. We pride ourselves on being an inclusive, equal opportunity workplace, committed to diversity and accessibility for all applicants.