Software Engineer, AI Infrastructure, Entry Level

Jobright is an AI-powered career platform that helps job seekers discover the top opportunities in the US. We are NOT a staffing agency. Jobright does not hire directly for these positions. We connect you with verified openings from employers you can trust.

Job Summary:

Robinhood is on a mission to democratize finance for all, and they are seeking a Software Engineer for their AI Infrastructure team. The role involves evolving the ML platform and services to support scalable, reliable, and secure model development and deployment, collaborating closely with Data Scientists and Applied ML Engineers.

Responsibilities:

• Design, build, and maintain scalable systems for deploying, monitoring, and managing Machine Learning models in production.

• Partner with ML practitioners to streamline workflows, integrate internal ML libraries, and optimize model performance.

• Contribute to the development and scaling of our feature store, enabling efficient feature retrieval across real-time and batch use cases.

• Implement robust observability for model performance, data pipelines, and feature freshness.

• Manage and optimize cloud compute resources (CPU/GPU) to support cost-effective training and inference across AWS.

Qualifications:

Required:

• 2+ years of software engineering experience, ideally within ML infrastructure, data engineering, or model operations.

• Hands-on experience with model serving, distributed systems, or production ML workflows.

• Familiarity with modern ML infrastructure tools (e.g., Ray, Kubeflow, SageMaker, TensorFlow Serving, Triton).

• Strong proficiency in Python, C++, or similar languages, and experience with ML frameworks like TensorFlow or PyTorch.

• Hands-on experience with large-scale search systems, including embedding models, vector databases, and distributed retrieval engines, using platforms like Qdrant, ChromaDB, Elasticsearch etc. with dense vector search capabilities.

• A collaborative mindset with an interest in building reliable platforms that support company-wide impact.

Company:

Robinhood is a stock brokerage that allows customers to buy and sell stocks, options, ETFs, and cryptocurrencies with zero commission. Founded in 2013, headquartered in Menlo Park, California, USA, team size 1001-5000 employees, currently Public Company.