AI Research Intern position focused on LLM agent research at Assent. Contribute to AI features enhancing supply chain sustainability in a remote-first environment.
Responsibilities
Research and prototype LLM-based agent architectures, exploring task decomposition, reasoning, and collaboration between agents
Implement and evaluate proof-of-concept AI systems using open-source or proprietary LLM models and frameworks.
Develop scalable experiment pipelines to assess model behavior, safety, and reliability
Build and optimize Python-based backend components, including prompt chains, vector stores, and evaluation tooling
Integrate prototypes with internal data systems and APIs, simulating realistic enterprise environments
Collaborate with ML engineers and product teams to define success metrics and transition promising prototypes toward production
Document findings in internal research reports, technical memos, and demonstrate the results
Requirements
Currently pursuing a PhD (3+ years completed) in Computer Science, Machine Learning, AI, or a related discipline
Strong coding and prototyping skills in Python, including familiarity with one or more of: PyTorch, TensorFlow, or similar ML frameworks
LLM agent frameworks (LangChain, LlamaIndex, DSPy, AutoGen, or Semantic Kernel)
Vector databases (FAISS, Chroma, Pinecone, or Milvus)
RAG and retrieval systems using embeddings and knowledge graphs
Deep understanding of LLM internals, including prompting strategies, fine-tuning, and model evaluation
Experience with cloud-based development (AWS) and modern MLOps or LLMOps workflows is an asset
Strong experimental design, data analysis, and communication skills
Benefits
vacation time that increases with tenure
comprehensive benefits packages (details vary by country)
life leave days
competitive base salary
corporate bonus program
retirement savings options
flexible work options
volunteer days
opportunities to get involved in corporate giving initiatives