AI/ML Engineer

Company DescriptionUltraShip, 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 DescriptionWe’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 ResponsibilitiesBuild AI-powered features using Python, PyTorch/TensorFlow, and modern ML librariesDevelop 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 platformsCreate evaluation frameworks for model accuracy and reliabilityMonitor models post-deployment and continuously improve themQualifications2–3 years of experience in AI/ML engineering roles1+ years experience with LLM frameworks like LangChain, LlamaIndex, or similar1+ years experience with NLP, recommendation systems, or structured prediction tasksFamiliarity with model training, fine-tuning, and inference optimizationUnderstanding of RAG, embeddings, prompt engineering, and evaluationHands-on experience working with structured + unstructured data pipelinesStrong foundation in Computer Science concepts including algorithms, data structures, and system design, Python programming and problem-solving abilityExperience building and deploying ML models beyond notebooks[Nice to have] Prior experience in a SaaS or B2B product environment

Apply Now