Timezone: »

Flashlight: Enabling Innovation in Tools for Machine Learning
Jacob Kahn · Vineel Pratap · Tatiana Likhomanenko · Qiantong Xu · Awni Hannun · Jeff Cai · Paden Tomasello · Ann Lee · Edouard Grave · Gilad Avidov · Benoit Steiner · Vitaliy Liptchinsky · Gabriel Synnaeve · Ronan Collobert

Thu Jul 21 11:45 AM -- 11:50 AM (PDT) @ Room 309

As the computational requirements for machine learning systems and the size and complexity of machine learning frameworks increases, essential framework innovation has become challenging. While computational needs have driven recent compiler, networking, and hardware advancements, utilization of those advancements by machine learning tools is occurring at a slower pace. This is in part due to the difficulties involved in prototyping new computational paradigms with existing frameworks. Large frameworks prioritize machine learning researchers and practitioners as end users and pay comparatively little attention to systems researchers who can push frameworks forward --- we argue that both are equally important stakeholders. We introduce Flashlight, an open-source library built to spur innovation in machine learning tools and systems by prioritizing open, modular, customizable internals and state-of-the-art, research-ready models and training setups across a variety of domains. Flashlight allows systems researchers to rapidly prototype and experiment with novel ideas in machine learning computation and has low overhead, competing with and often outperforming other popular machine learning frameworks. We see Flashlight as a tool enabling research that can benefit widely used libraries downstream and bring machine learning and systems researchers closer together.

Author Information

Jacob Kahn (Facebook AI Research (FAIR))
Vineel Pratap (Facebook)
Tatiana Likhomanenko (Apple)
Qiantong Xu (Facebook AI Research)
Awni Hannun (Zoom AI)
Jeff Cai (None)
Paden Tomasello (Facebook AI Research)
Ann Lee (Facebook, Inc.)
Edouard Grave (Facebook AI Research)
Gilad Avidov (Facebook)
Benoit Steiner (FAIR)
Vitaliy Liptchinsky (Facebook)
Gabriel Synnaeve (Facebook AI Research)
Ronan Collobert (Facebook AI Research)

Related Events (a corresponding poster, oral, or spotlight)

More from the Same Authors