Python Backend Developer, MLOps Engineer

Summary

Join Prompt as a Python Backend Developer / MLOps Engineer and play a crucial role in enhancing our AI infrastructure and python backend systems. You will ensure that our machine learning models and use-cases are efficiently developed, deployed, and managed.

Requirements

• Bachelor's degree in Computer Science, Engineering, or a related field; advanced degree preferred
• Proven experience in MLOps, machine learning model deployment, and API development. Experience with FastAPI and Streamlit preferred
• Strong knowledge of cloud services, particularly AWS (EC2, ECS, Lambda, SageMaker)
• Proficiency in programming languages such as Python and familiarity with frameworks for data science and machine learning
• Experience with database management and migration processes, experience with SQLAlchemy and Alembic preferred
• Solid understanding of software development best practices, including version control and testing in Python. Familiarity with git and pytest required; familiarity with Gitlab CI preferred

Responsibilities

• Architect and refine our core Python packages for AI development and deployment
• Enhance and maintain the AI frontend to facilitate rapid prototyping and demo development for AI engineers
• Develop and refine an AI management dashboard for internal monitoring of AI assets and systems and as a UI for internal tooling
• Refactor existing ML operations utilities to improve functionality and support connections between modeling/applications and our AI data plane
• Improve and maintain the AI API to serve out AI use-cases to downstream stakeholders, focusing on security and scalability
• Improve test coverage over the codebase
• Research and implement cost-effective deployment strategies for various ETL pipelines and ML models, optimizing for specific use cases
• Review and manage AWS resources and associated costs for the AI team, ensuring efficient utilization
• Develop and integrate interfaces for third-party software and packages
• Oversee the maintenance and integrity of various AI environments, ensuring smooth transitions and database management
• Setup and maintain CI/CD pipelines for AI models and applications

Benefits

• Competitive salaries
• Remote/hybrid environment
• Potential equity compensation for outstanding performance
• Flexible PTO
• Company-wide sponsored lunches
• Company paid disability and life insurance benefits
• Company paid family and medical leave
• Medical, dental, and vision insurance benefits
• Discounted pet insurance
• FSA/DCA and commuter benefits
• 401k