English
Related papers

Related papers: PyTorchPipe: a framework for rapid prototyping of …

200 papers

We introduce WordScape, a novel pipeline for the creation of cross-disciplinary, multilingual corpora comprising millions of pages with annotations for document layout detection. Relating visual and textual items on document pages has…

Probabilistic graphical models (PGMs) serve as a powerful framework for modeling complex systems with uncertainty and extracting valuable insights from data. However, users face challenges when applying PGMs to their problems in terms of…

Machine Learning · Computer Science 2024-05-29 Jiantong Jiang , Zeyi Wen , Peiyu Yang , Atif Mansoor , Ajmal Mian

Software development is shifting from traditional programming to AI-integrated applications that leverage generative AI and large language models (LLMs) during runtime. However, integrating LLMs remains complex, requiring developers to…

Image pyramids are commonly used in modern computer vision tasks to obtain multi-scale features for precise understanding of images. However, image pyramids process multiple resolutions of images using the same large-scale model, which…

Computer Vision and Pattern Recognition · Computer Science 2024-10-29 Xizhou Zhu , Xue Yang , Zhaokai Wang , Hao Li , Wenhan Dou , Junqi Ge , Lewei Lu , Yu Qiao , Jifeng Dai

Over the past decade, data science and machine learning has grown from a mysterious art form to a staple tool across a variety of fields in academia, business, and government. In this paper, we introduce the concept of tree-based pipeline…

Machine Learning · Computer Science 2016-02-01 Randal S. Olson , Ryan J. Urbanowicz , Peter C. Andrews , Nicole A. Lavender , La Creis Kidd , Jason H. Moore

Training large language models (LLMs) now requires resources that exceed a single datacenter, making cross-datacenter strategies increasingly crucial. We present CrossPipe, a framework designed to optimize model training across…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-07-02 Tiancheng Chen , Ales Kubicek , Langwen Huang , Torsten Hoefler

PyTorch \texttt{2.x} introduces a compiler designed to accelerate deep learning programs. However, for machine learning researchers, adapting to the PyTorch compiler to full potential can be challenging. The compiler operates at the Python…

Machine Learning · Computer Science 2024-03-22 Kaichao You , Runsheng Bai , Meng Cao , Jianmin Wang , Ion Stoica , Mingsheng Long

Flow Matching (FM) is a recent framework for generative modeling that has achieved state-of-the-art performance across various domains, including image, video, audio, speech, and biological structures. This guide offers a comprehensive and…

Python has become the de-facto language for training deep neural networks, coupling a large suite of scientific computing libraries with efficient libraries for tensor computation such as PyTorch or TensorFlow. However, when models are used…

Machine Learning · Computer Science 2021-04-02 Zachary DeVito , Jason Ansel , Will Constable , Michael Suo , Ailing Zhang , Kim Hazelwood

NLP Workbench is a web-based platform for text mining that allows non-expert users to obtain semantic understanding of large-scale corpora using state-of-the-art text mining models. The platform is built upon latest pre-trained models and…

Computation and Language · Computer Science 2024-03-06 Peiran Yao , Matej Kosmajac , Abeer Waheed , Kostyantyn Guzhva , Natalie Hervieux , Denilson Barbosa

Text-to-image generation has evolved beyond single monolithic models to complex multi-component pipelines. These combine fine-tuned generators, adapters, upscaling blocks and even editing steps, leading to significant improvements in image…

Computer Vision and Pattern Recognition · Computer Science 2025-11-04 Uri Gadot , Rinon Gal , Yftah Ziser , Gal Chechik , Shie Mannor

Serving Large Language Models (LLMs) in production faces significant challenges from highly variable request patterns and severe resource fragmentation in serverless clusters. Current systems rely on static pipeline configurations that…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-12 Yanying Lin , Shijie Peng , Chengzhi Lu , Chengzhong Xu , Kejiang Ye

The parameter size of modern large language models (LLMs) can be scaled up via the sparsely-activated Mixture-of-Experts (MoE) technique to avoid excessive increase of the computational costs. To further improve training efficiency,…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-08 Yunqi Gao , Bing Hu , Mahdi Boloursaz Mashhadi , A-Long Jin , Yanfeng Zhang , Pei Xiao , Rahim Tafazolli , Merouane Debbah

This paper presents TorchNWP, a compilation library tool for the efficient coupling of artificial intelligence components and traditional numerical models. It aims to address the issues of poor cross-language compatibility, insufficient…

Mathematical Software · Computer Science 2026-03-19 Sa Xiao , Hao Jing , Honglu Sun , Haoyu Li

Training data is an essential resource for creating capable and robust vision systems which are integral to the proper function of many robotic systems. Synthesized training data has been shown in recent years to be a viable alternative to…

Robotics · Computer Science 2024-11-12 Peter Gavriel , Adam Norton , Kenneth Kimble , Megan Zimmerman

Image pyramids are widely adopted in top-performing methods to obtain multi-scale features for precise visual perception and understanding. However, current image pyramids use the same large-scale model to process multiple resolutions of…

Computer Vision and Pattern Recognition · Computer Science 2025-07-28 Zhaokai Wang , Xizhou Zhu , Xue Yang , Gen Luo , Hao Li , Changyao Tian , Wenhan Dou , Junqi Ge , Lewei Lu , Yu Qiao , Jifeng Dai

Throughout the last years, machine learning techniques have been broadly encouraged in the context of deep learning architectures. An exciting algorithm denoted as Restricted Boltzmann Machine relies on energy- and probabilistic-based…

Machine Learning · Computer Science 2020-09-24 Mateus Roder , Gustavo Henrique de Rosa , João Paulo Papa

Interventions on model-internal states are fundamental operations in many areas of AI, including model editing, steering, robustness, and interpretability. To facilitate such research, we introduce $\textbf{pyvene}$, an open-source Python…

In this paper, we present PARTIME, a software library written in Python and based on PyTorch, designed specifically to speed up neural networks whenever data is continuously streamed over time, for both learning and inference. Existing…

Machine Learning · Computer Science 2022-12-05 Enrico Meloni , Lapo Faggi , Simone Marullo , Alessandro Betti , Matteo Tiezzi , Marco Gori , Stefano Melacci

Despite the prevalence of symmetry in scientific linear systems, these structural properties are often underutilized by standard computational software. This paper introduces PySymmetry, an open-source Sage/Python framework that implements…

Group Theory · Mathematics 2025-09-25 Leon D. da Silva , Marcelo P. Santos