English
Related papers

Related papers: TorchOptics: An open-source Python library for dif…

200 papers

Multiple Instance Learning (MIL) is a powerful framework for weakly supervised learning, particularly useful when fine-grained annotations are unavailable. Despite growing interest in deep MIL methods, the field lacks standardized tools for…

Despite significant progress of applying deep learning methods to the field of content-based image retrieval, there has not been a software library that covers these methods in a unified manner. In order to fill this gap, we introduce…

Information Retrieval · Computer Science 2020-08-06 Benyi Hu , Ren-Jie Song , Xiu-Shen Wei , Yazhou Yao , Xian-Sheng Hua , Yuehu Liu

Developing optical systems for free-space applications requires simulation tools that accurately capture turbulence-induced wavefront distortions and support gradient-based optimization. Here we introduce TurPy, a GPU-accelerated, fully…

We present WOFRY (Wave Optics FRamework in pYthon), a specialized toolbox designed for wave optics modeling, with particular emphasis on partial coherence. This package is tailored to assist synchrotron scientists and engineers in the…

Deep learning algorithms have made many breakthroughs and have various applications in real life. Computational resources become a bottleneck as the data and complexity of the deep learning pipeline increases. In this paper, we propose…

Machine Learning · Computer Science 2021-05-05 Salman Ahmed , Hammad Naveed

Differentiable sparse linear algebra is foundational for scientific machine learning, yet PyTorch lacks a unified library for it: \texttt{torch.sparse} provides only low-level kernels and a non-differentiable, CPU-only \texttt{spsolve}, and…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-07 Mingyuan Chi , Shizheng Wen

TorchSurv is a Python package that serves as a companion tool to perform deep survival modeling within the PyTorch environment. Unlike existing libraries that impose specific parametric forms, TorchSurv enables the use of custom…

Machine Learning · Computer Science 2024-04-18 Mélodie Monod , Peter Krusche , Qian Cao , Berkman Sahiner , Nicholas Petrick , David Ohlssen , Thibaud Coroller

Deep learning has brought significant advancements to X-ray Computed Tomography (CT) reconstruction, offering solutions to challenges arising from modern imaging technologies. These developments benefit from methods that combine classical…

Machine Learning · Computer Science 2025-11-12 Linda-Sophie Schneider , Yipeng Sun , Chengze Ye , Markus Michen , Andreas Maier

Fine-Grained Image Recognition (FGIR) is a fundamental and challenging task in computer vision and multimedia that plays a crucial role in Intellectual Economy and Industrial Internet applications. However, the absence of a unified…

Computer Vision and Pattern Recognition · Computer Science 2023-11-27 Jiabei He , Yang Shen , Xiu-Shen Wei , Ye Wu

The KeOps library provides a fast and memory-efficient GPU support for tensors whose entries are given by a mathematical formula, such as kernel and distance matrices. KeOps alleviates the major bottleneck of tensor-centric libraries for…

Machine Learning · Computer Science 2021-04-10 Benjamin Charlier , Jean Feydy , Joan Alexis Glaunès , François-David Collin , Ghislain Durif

We present an open-source tensor network Python library for quantum many-body simulations. At its core is an abelian-symmetric tensor, implemented as a sparse block structure managed by logical layer on top of dense multi-dimensional array…

Strongly Correlated Electrons · Physics 2025-03-05 Marek M. Rams , Gabriela Wójtowicz , Aritra Sinha , Juraj Hasik

We present NeuralMag, a flexible and high-performance open-source Python library for micromagnetic simulations. NeuralMag leverages modern machine learning frameworks, such as PyTorch and JAX, to perform efficient tensor operations on…

Deep learning has significantly improved 2D image recognition. Extending into 3D may advance many new applications including autonomous vehicles, virtual and augmented reality, authoring 3D content, and even improving 2D recognition.…

Computer Vision and Pattern Recognition · Computer Science 2020-07-17 Nikhila Ravi , Jeremy Reizenstein , David Novotny , Taylor Gordon , Wan-Yen Lo , Justin Johnson , Georgia Gkioxari

Astronomical radio interferometers achieve exquisite angular resolution by cross-correlating signal from a cosmic source simultaneously observed by distant pairs of radio telescopes to produce a Fourier-type measurement called a visibility.…

Instrumentation and Methods for Astrophysics · Physics 2025-02-04 Ian Czekala , Jeff Jennings , Brianna Zawadzki , Kadri Nizam , Ryan Loomis , Megan Delamer , Kaylee de Soto , Robert Frazier , Hannah Grzybowski , Jane Huang , Mary Ogborn , Tyler Quinn

We present PyTorch Frame, a PyTorch-based framework for deep learning over multi-modal tabular data. PyTorch Frame makes tabular deep learning easy by providing a PyTorch-based data structure to handle complex tabular data, introducing a…

Machine Learning · Computer Science 2024-12-17 Weihua Hu , Yiwen Yuan , Zecheng Zhang , Akihiro Nitta , Kaidi Cao , Vid Kocijan , Jinu Sunil , Jure Leskovec , Matthias Fey

We describe a software package, TomOpt, developed to optimise the geometrical layout and specifications of detectors designed for tomography by scattering of cosmic-ray muons. The software exploits differentiable programming for the…

TenCirChem is an open-source Python library for simulating variational quantum algorithms for quantum computational chemistry. TenCirChem shows high performance on the simulation of unitary coupled-cluster circuits, using compact…

Achieving the desired optical response from a multilayer thin-film structure over a broad range of wavelengths and angles of incidence can be challenging. An advanced thin-film structure can consist of multiple materials with different…

Computational Physics · Physics 2022-05-25 Alexander Luce , Ali Mahdavi , Florian Marquardt , Heribert Wankerl

The goal of hyperparameter tuning (or hyperparameter optimization) is to optimize the hyperparameters to improve the performance of the machine or deep learning model. spotPython (``Sequential Parameter Optimization Toolbox in Python'') is…

Machine Learning · Computer Science 2023-06-08 Thomas Bartz-Beielstein

Deep learning has enabled major advances in the fields of computer vision, natural language processing, and multimedia among many others. Developing a deep learning system is arduous and complex, as it involves constructing neural network…

Machine Learning · Computer Science 2017-08-04 Hao Dong , Akara Supratak , Luo Mai , Fangde Liu , Axel Oehmichen , Simiao Yu , Yike Guo