← Search

Louis Serrano

8 accepted papers

2026

Test-time Generalization for Physics through Neural Operator Splitting

ICML 2026poster

Neural operators have shown promise in learning solution maps of partial differential equations (PDEs), but they often struggle to generalize when test inputs lie outside the training distribution, such as novel initial conditions, unseen PDE coefficients or unseen physics. Prior works address this …

Cited by 0SourceScholar
2025

ACCO: Accumulate While You Communicate for Communication-Overlapped Sharded LLM Training

NeurIPS 2025poster

Training LLMs relies on distributed implementations using multiple GPUs to compute gradients in parallel with sharded optimizers. However, synchronizing gradients in data parallel setups introduces communication overhead that grows with the number of workers, limiting parallelization efficiency. Loc…

Cited by 0SourcecodeScholar
2025

ENMA: Tokenwise Autoregression for Continuous Neural PDE Operators

NeurIPS 2025spotlight

Solving time-dependent parametric partial differential equations (PDEs) remains a fundamental challenge for neural solvers, particularly when generalizing across a wide range of physical parameters and dynamics. When data is uncertain or incomplete—as is often the case—a natural approach is to turn…

Cited by 0SourceScholar
2025

JAFAR: Jack up Any Feature at Any Resolution

NeurIPS 2025poster

Foundation Vision Encoders have become indispensable across a wide range of dense vision tasks. However, their operation at low spatial feature resolutions necessitates subsequent feature decompression to enable full-resolution processing. To address this limitation, we introduce JAFAR, a lightweigh…

Cited by 0SourcecodeScholar
2025

Learning a Neural Solver for Parametric PDEs to Enhance Physics-Informed Methods

ICLR 2025poster

Physics-informed deep learning often faces optimization challenges due to the complexity of solving partial differential equations (PDEs), which involve exploring large solution spaces, require numerous iterations, and can lead to unstable training. These challenges arise particularly from the ill-c…

Cited by 2SourcePDFScholar
2025

Zebra: In-Context Generative Pretraining for Solving Parametric PDEs

ICML 2025poster

Solving time-dependent parametric partial differential equations (PDEs) is challenging for data-driven methods, as these models must adapt to variations in parameters such as coefficients, forcing terms, and initial conditions. State-of-the-art neural surrogates perform adaptation through gradient-b…

Cited by 2SourcePDFScholar
2024

AROMA: Preserving Spatial Structure for Latent PDE Modeling with Local Neural Fields

NeurIPS 2024poster

We present AROMA (Attentive Reduced Order Model with Attention), a framework designed to enhance the modeling of partial differential equations (PDEs) using local neural fields. Our flexible encoder-decoder architecture can obtain smooth latent representations of spatial physical fields from a varie…

2023

Operator Learning with Neural Fields: Tackling PDEs on General Geometries

NeurIPS 2023poster

Machine learning approaches for solving partial differential equations require learning mappings between function spaces. While convolutional or graph neural networks are constrained to discretized functions, neural operators present a promising milestone toward mapping functions directly. Despite i…