AI/ML Engineer

Company Description

UltraShip, established in 2022, is a SaaS-based company focused on revolutionizing the logistics and supply chain industry through innovative technology. By integrating advanced AI solutions into a single interface, UltraShip TMS creates seamless experiences for 3PLs. Headquartered in the San Francisco Bay Area, UltraShip TMS is driving the transformation of logistics with technology-focused solutions designed for maximum synchronization and innovation.

Role Description

We’re looking for an AI/ML Engineer who thrives in a fast-moving startup environment, someone who can build, ship, iterate, and take ownership of real AI-powered products. This is not a research-only role. You will work on practical applications of machine learning and generative AI, deploying models into production and directly impacting customers. You will build data pipelines for training, evaluation, and inference while optimizing model performance, cost, and latency in real-world systems. Don't think twice to Experiment rapidly and turn prototypes into scalable AI services.

If you love solving messy problems, working end-to-end, and building AI systems that actually ship, this role is for you. The candidate will collaborate with cross-functional teams to deliver high-impact solutions, and contribute to the overall improvement of the Product.

Position: This is a full-time, 100% remote-first role in India.

Key Responsibilities

• Build AI-powered features using Python, PyTorch/TensorFlow, and modern ML libraries
• Develop complex multi-step agentic pipelines and implement workflows around LLMs (OpenAI, Anthropic, open-source models etc)
• Build tools enabling AIs to interact with critical enterprise software platforms.
• Develop retrieval systems using vector databases (Pinecone etc)
• Deploy AI services using FastAPI, Docker, Kubernetes, or cloud platforms
• Create evaluation frameworks for model accuracy and reliability
• Monitor models post-deployment and continuously improve them

Qualifications

• 2–3 years of experience in AI/ML engineering roles
• 1+ years experience with LLM frameworks like LangChain, LlamaIndex, or similar
• 1+ years experience with NLP, recommendation systems, or structured prediction tasks
• Familiarity with model training, fine-tuning, and inference optimization
• Understanding of RAG, embeddings, prompt engineering, and evaluation
• Hands-on experience working with structured + unstructured data pipelines
• Strong foundation in Computer Science concepts including algorithms, data structures, and system design, Python programming and problem-solving ability
• Experience building and deploying ML models beyond notebooks
• [Nice to have] Prior experience in a SaaS or B2B product environment