Machine Learning Engineer

RE Partners • New York, NY

2d

About The Position

Machine Learning Engineer

Requirements

• Strong Python skills, with experience in ML libraries such as scikit-learn, PyTorch, TensorFlow.
• Hands-on experience with real-time time-series data processing (e.g., Kafka, Flink, Spark Streaming).
• Proven experience in time-series modeling: LSTM, Prophet, Transformer-based architectures.
• Experience with unsupervised learning and clustering techniques (DBSCAN, K-Means).
• Familiarity with LLM APIs (OpenAI, Anthropic, Gemini) and prompt engineering best practices.
• Understanding of prompt caching and embedding-based retrieval techniques.
• Solid MLOps experience on AWS (SageMaker, Lambda, ECS, CloudWatch, Step Functions).
• Experience with model quantization, pruning, ONNX conversion, and other techniques for mobile deployment.

Nice To Haves

• Experience with vector databases (e.g., Pinecone, FAISS, Weaviate).
• Exposure to edge AI toolkits (e.g., TensorFlow Lite, Core ML, MediaPipe).
• Understanding of data privacy and security considerations in streaming ML applications.

Responsibilities

• Develop and maintain machine learning models for time-series forecasting, anomaly detection, and clustering in real-time environments.
• Integrate Large Language Models (LLMs) for task-specific use cases, including prompt optimization and caching strategies.
• Design and implement end-to-end MLOps pipelines on AWS, including model training, validation, deployment, and monitoring.
• Optimize models for edge and mobile execution with a focus on performance, battery, and resource constraints.
• Collaborate with data engineers to handle high-velocity streaming data pipelines.
• Evaluate and improve prompt quality, retrieval techniques, and contextual embeddings for LLM workflows.
• Ensure scalability, reliability, and real-time performance of deployed ML systems.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: No Education Listed
• Number of Employees: 251-500 employees