Lattice AI Research

Research focus

Robotics & Embodied AI (1)World Models & Planning (1)Architecture Design (Transformers, SSMs, MoE) (1)Recommendation & Information Retrieval (1)

Frequent co-authors

Arnav Kumar Jain (1)Yilin Wu (1)Yilin Wu (1)Jesse Farebrother (1)

Papers (2)

Jun 11, 2026

MilaJun 11, 2026·also CMU ML

$\texttt{WEAVER}$, Better, Faster, Longer: An Effective World Model for Robotic Manipulation

Achieving a 38% boost in real-world success rates for robotic manipulation, $\texttt{WEAVER}$ redefines the capabilities of world models in robotics.

Arnav Kumar Jain, A. Jain, Yilin Wu +7

Robotics & Embodied AI World Models & Planning

Apr 1, 2025

Apr 1, 2025·also Meta AI, Mila, Stanford HAI, Edinburgh +4

Command A: An Enterprise-Ready Large Language Model

Command A shows how to build an enterprise-grade LLM that balances performance, efficiency, and multilingual capabilities using decentralized training and model merging.

Team Cohere, Aakanksha, Arash Ahmadian +22338

Architecture Design (Transformers, SSMs, MoE)Recommendation & Information Retrieval Tool Use & Agents

Search

A. Jain

Publication activitypapers/week, last 8 weeks

Research focus

Frequent co-authors

Papers (2)