KineticXGPU: A Tensorized Collision Operator for Dark-Sector Self-Scattering
Pith reviewed 2026-07-02 10:24 UTC · model grok-4.3
The pith
A PyTorch tensor contraction computes the 2-to-2 elastic collision operator and shows self-interactions erase bimodal dark-sector distributions.
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The discretized 2-to-2 elastic self-collision operator for dark-sector momentum distributions can be expressed as tensor contractions that run efficiently on GPUs; when this operator is included in a two-source freeze-in scenario, increasing the strength of elastic self-interactions erases the bimodal structure and drives the distribution toward a Maxwell-Boltzmann distribution.
What carries the argument
The tensorized discretization of the 2-to-2 elastic collision operator, implemented as PyTorch tensor contractions.
If this is right
- Self-interactions of sufficient strength can be expected to thermalize any initial non-equilibrium features in dark-sector momentum distributions.
- The phase-space results can be cross-checked against the simpler fluid equations for number density and velocity dispersion.
- GPU tensor contractions provide a measurable runtime advantage over CPU implementations for repeated evaluations of the collision operator.
- The same tensorized formulation can be reused for other dark-sector interaction strengths or initial conditions.
Where Pith is reading between the lines
- The method could be adapted to study inelastic or number-changing processes by extending the tensor contraction structure.
- If the thermalization effect holds across a range of models, cosmological observables sensitive to dark-sector velocity dispersion might indirectly constrain the self-interaction strength.
- Direct comparison of the code output against known analytic equilibrium solutions would test the numerical fidelity of the tensor contractions.
Load-bearing premise
The tensorized discretization reproduces the physical 2-to-2 collision operator without numerical artifacts that would alter the reported change in distribution shape.
What would settle it
Running an independent, non-tensorized numerical integration of the same Boltzmann equation for identical parameters and finding that the final distribution shape differs from the GPU result.
read the original abstract
In this work, we present KineticXGPU, a PyTorch-based implementation of the $2\to 2$ elastic self-collision operator for dark-sector momentum distributions. The discretized collision operator can be expressed as tensor contractions and is therefore well suited for GPUs. As an application, we study a two-source freeze-in scenario in which the final distribution can develop a bimodal shape. We show that increasing the strength of elastic self-interactions progressively erases this structure and drives the distribution toward a Maxwell-Boltzmann distribution. We compare the phase-space formulation with a set of fluid equations that couple the number density and velocity dispersion. We also compare CPU and GPU runtimes and demonstrate the computational advantage of the tensorized approach. The code is publicly available on GitHub.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The manuscript introduces KineticXGPU, a PyTorch-based implementation of the 2→2 elastic self-collision operator for dark-sector momentum distributions. The discretized operator is formulated as tensor contractions to enable efficient GPU computation. As an application, the authors consider a two-source freeze-in scenario that produces a bimodal momentum distribution and demonstrate that increasing the strength of elastic self-interactions erases the bimodality, driving the distribution toward a Maxwell-Boltzmann form. The work also compares the phase-space evolution to a set of fluid equations coupling number density and velocity dispersion, benchmarks CPU versus GPU runtimes, and releases the code publicly on GitHub.
Significance. If the numerical implementation is shown to be accurate, the tensorized approach offers a practical tool for evolving non-thermal distributions under self-interactions, which is relevant for dark-sector model building. The public code release is a clear strength that enables reproducibility and community use.
major comments (1)
- [application section] Application section: The central claim that stronger elastic self-interactions erase the bimodal structure and drive the distribution to Maxwell-Boltzmann rests on the numerical evolution of the discretized collision operator. No tests are presented demonstrating that the tensorized discretization conserves particle number and energy to machine precision, converges with momentum-grid resolution, or reproduces known analytic limits (e.g., thermalization rate in the weak-coupling regime). Without such checks, it is impossible to rule out that the reported shape change arises from numerical artifacts rather than the physical operator.
Simulated Author's Rebuttal
We thank the referee for their careful reading of the manuscript and for highlighting the importance of numerical validation. We address the single major comment below and will incorporate the requested tests in the revised version.
read point-by-point responses
-
Referee: The central claim that stronger elastic self-interactions erase the bimodal structure and drive the distribution to Maxwell-Boltzmann rests on the numerical evolution of the discretized collision operator. No tests are presented demonstrating that the tensorized discretization conserves particle number and energy to machine precision, converges with momentum-grid resolution, or reproduces known analytic limits (e.g., thermalization rate in the weak-coupling regime). Without such checks, it is impossible to rule out that the reported shape change arises from numerical artifacts rather than the physical operator.
Authors: We agree that explicit validation is necessary to support the central claim. The current manuscript does not contain dedicated tests for exact conservation of particle number and energy, grid convergence, or reproduction of the weak-coupling thermalization rate. In the revised version we will add a new subsection (likely in Section 3 or an appendix) that reports: (i) relative conservation errors for number and energy to machine precision across multiple time steps, (ii) results for successively refined momentum grids demonstrating convergence of the final distribution shape, and (iii) a direct comparison of the numerically extracted thermalization timescale against the analytic expectation in the weak-coupling regime. These additions will allow readers to assess that the reported erasure of bimodality is driven by the physical operator. revision: yes
Circularity Check
No circularity; implementation and application results are independent of inputs by construction.
full rationale
The paper describes a PyTorch tensorized discretization of the 2→2 elastic collision operator and applies it to evolve a two-source freeze-in distribution under varying self-interaction strengths. No derivation chain reduces a claimed prediction or first-principles result to fitted parameters, self-definitions, or self-citations that bear the central load. The reported shape evolution follows directly from numerically integrating the discretized operator on a momentum grid; the discretization itself is presented as an independent computational encoding rather than an ansatz or renaming that presupposes the Maxwell-Boltzmann outcome. The comparison to fluid equations and runtime benchmarks further stands on external verification of the code, not internal reduction. This is a standard self-contained implementation paper with no load-bearing circular steps.
Axiom & Free-Parameter Ledger
Reference graph
Works this paper leans on
-
[1]
Silk et al.,Particle Dark Matter: Observations, Models and Searches, Cambridge Univ
J. Silk et al.,Particle Dark Matter: Observations, Models and Searches, Cambridge Univ. Press, Cambridge (2010), 10.1017/CBO9780511770739
-
[2]
M. Cirelli, A. Strumia and J. Zupan,Dark Matter,2406.01705
work page internal anchor Pith review Pith/arXiv arXiv
-
[3]
G. Arcadi, D. Cabo-Almeida, M. Dutra, P. Ghosh, M. Lindner, Y. Mambrini et al.,The Waning of the WIMP: Endgame?,Eur. Phys. J. C85(2025) 152 [2403.15860]. [4]LZcollaboration,Dark Matter Search Results from 4.2 Tonne-Years of Exposure of the LUX-ZEPLIN (LZ) Experiment,Phys. Rev. Lett.135(2025) 011802 [2410.17036]. [5]LZcollaboration,Searches for Light Dark ...
-
[4]
L.J. Hall, K. Jedamzik, J. March-Russell and S.M. West,Freeze-In Production of FIMP Dark Matter,JHEP03(2010) 080 [0911.1120]
work page internal anchor Pith review Pith/arXiv arXiv 2010
-
[5]
J.L. Feng, M. Kaplinghat and H.-B. Yu,Sommerfeld Enhancements for Thermal Relic Dark Matter,Phys. Rev. D82(2010) 083525 [1005.4678]
work page internal anchor Pith review Pith/arXiv arXiv 2010
- [6]
-
[7]
Observational and Theoretical Constraints on Singular Dark Matter Halos
R.A. Flores and J.R. Primack,Observational and theoretical constraints on singular dark matter halos,Astrophys. J. Lett.427(1994) L1 [astro-ph/9402004]
work page internal anchor Pith review Pith/arXiv arXiv 1994
-
[8]
Moore,Evidence against dissipationless dark matter from observations of galaxy haloes, Nature370(1994) 629
B. Moore,Evidence against dissipationless dark matter from observations of galaxy haloes, Nature370(1994) 629
1994
-
[9]
S.-H. Oh, C. Brook, F. Governato, E. Brinks, L. Mayer, W.J.G. de Blok et al.,The Central Slope of Dark Matter Cores in Dwarf Galaxies: Simulations versus THINGS,Astron. J.142 (2011) 24 [1011.2777]
work page internal anchor Pith review Pith/arXiv arXiv 2011
-
[10]
A Method for Measuring (Slopes of) the Mass Profiles of Dwarf Spheroidal Galaxies
M.G. Walker and J. Pe˜ narrubia,A Method for Measuring (Slopes of) the Mass Profiles of Dwarf Spheroidal Galaxies,The Astrophysical Journal742(2011) 20 [1108.2404]
work page internal anchor Pith review Pith/arXiv arXiv 2011
-
[11]
Too big to fail? The puzzling darkness of massive Milky Way subhaloes
M. Boylan-Kolchin, J.S. Bullock and M. Kaplinghat,Too big to fail? The puzzling darkness of massive Milky Way subhaloes,Mon. Not. Roy. Astron. Soc.415(2011) L40 [1103.0007]
work page internal anchor Pith review Pith/arXiv arXiv 2011
-
[12]
Carlson, M.E
E.D. Carlson, M.E. Machacek and L.J. Hall,Self-interacting dark matter,Astrophys. J.398 (1992) 43
1992
-
[13]
Constraints on Self-Interacting Dark Matter
A.A. de Laix, R.J. Scherrer and R.K. Schaefer,Constraints of selfinteracting dark matter, Astrophys. J.452(1995) 495 [astro-ph/9502087]. – 22 –
work page internal anchor Pith review Pith/arXiv arXiv 1995
-
[14]
Y. Hochberg, E. Kuflik, T. Volansky and J.G. Wacker,Mechanism for Thermal Relic Dark Matter of Strongly Interacting Massive Particles,Phys. Rev. Lett.113(2014) 171301 [1402.5143]
work page internal anchor Pith review Pith/arXiv arXiv 2014
-
[15]
Cannibal Dark Matter and Large Scale Structure
M.A. Buen-Abad, R. Emami and M. Schmaltz,Cannibal Dark Matter and Large Scale Structure,Phys. Rev. D98(2018) 083517 [1803.08062]
work page internal anchor Pith review Pith/arXiv arXiv 2018
-
[16]
Dark Matter Self Interactions and its Impact on Large Scale Structures
C. Chatterjee,Dark Matter Self Interactions and its Impact on Large Scale Structures, 1901.05658
work page internal anchor Pith review Pith/arXiv arXiv 1901
-
[17]
A.L. Erickcek, P. Ralegankar and J. Shelton,Cannibal domination and the matter power spectrum,Phys. Rev. D103(2021) 103508 [2008.04311]
-
[18]
S. Heimersheim, N. Sch¨ oneberg, D.C. Hooper and J. Lesgourgues,Cannibalism hinders growth: Cannibal Dark Matter and theS 8 tension,JCAP12(2020) 016 [2008.08486]
-
[19]
M. Hufnagel and M.H.G. Tytgat,The domain of a cannibal dark matter,JCAP09(2023) 012 [2212.09759]
-
[20]
Freeze-in production of decaying dark matter in five steps
S. Heeba, F. Kahlhoefer and P. St¨ ocker,Freeze-in production of decaying dark matter in five steps,JCAP11(2018) 048 [1809.04849]
work page internal anchor Pith review Pith/arXiv arXiv 2018
-
[21]
J. March-Russell, H. Tillim and S.M. West,Reproductive freeze-in of self-interacting dark matter,Phys. Rev. D102(2020) 083018 [2007.14688]
-
[22]
N. Bernal and X. Chu,Z 2 SIMP Dark Matter,JCAP01(2016) 006 [1510.08527]
work page internal anchor Pith review Pith/arXiv arXiv 2016
-
[23]
E. Cervantes and A. Hryczuk,Freezing-in cannibal dark sectors,JHEP11(2024) 050 [2407.12104]
- [24]
-
[25]
S.W. Randall, M. Markevitch, D. Clowe, A.H. Gonzalez and M. Bradac,Constraints on the Self-Interaction Cross-Section of Dark Matter from Numerical Simulations of the Merging Galaxy Cluster 1E 0657-56,Astrophys. J.679(2008) 1173 [0704.0261]
work page internal anchor Pith review Pith/arXiv arXiv 2008
-
[26]
A. Hryczuk and M. Laletin,Impact of dark matter self-scattering on its relic abundance,Phys. Rev. D106(2022) 023007 [2204.07078]
-
[27]
S. Chatterjee and A. Hryczuk,Conversions in two-component dark sectors: a phase space level analysis,JHEP07(2025) 279 [2502.08725]
-
[28]
Exploring non-equilibrium effects in sequential freeze-in
S. Chatterjee and A. Hryczuk,Exploring non-equilibrium effects in sequential freeze-in, 2604.14688
work page internal anchor Pith review Pith/arXiv arXiv
- [29]
-
[30]
Novel constraints on non-cold (non-thermal) Dark Matter from Lyman-$\alpha$ forest data
R. Murgia, V. Irˇ siˇ c and M. Viel,Novel constraints on noncold, nonthermal dark matter from Lyman-αforest data,Phys. Rev. D98(2018) 083540 [1806.08371]
work page internal anchor Pith review Pith/arXiv arXiv 2018
-
[31]
F. D’Eramo and A. Lenoci,Lower mass bounds on FIMP dark matter produced via freeze-in, JCAP10(2021) 045 [2012.01446]
- [32]
- [33]
- [34]
-
[35]
F. D’Eramo, A. Lenoci and A. Dekker,Dark matter freeze-in and small-scale observables: Novel mass bounds and viable particle candidates,Phys. Rev. D112(2025) 116008 [2506.13864]
-
[36]
S.-Y. Zhao, Y.-C. Dai, W. Liao and Y.-S. Lu,Lyman-αForest Constraint on Dark Matter from Dark Sector Decay,2603.24331
-
[37]
K.R. Dienes, J.N. Howard, F. Huang, Y.-Z. Li and B. Thomas,Machine Learning Does It and Does It Better: Unearthing Primordial Dark-Matter Velocities from the Matter Power Spectrum,2606.13527
work page internal anchor Pith review Pith/arXiv arXiv
- [38]
-
[39]
The Evolution of the Intergalactic Medium
M. McQuinn,The Evolution of the Intergalactic Medium,Ann. Rev. Astron. Astrophys.54 (2016) 313 [1512.00086]
work page internal anchor Pith review Pith/arXiv arXiv 2016
-
[40]
Huo,Matter Power Spectrum of Light Freeze-in Dark Matter: With or without Self-Interaction,Phys
R. Huo,Matter Power Spectrum of Light Freeze-in Dark Matter: With or without Self-Interaction,Phys. Lett. B802(2020) 135251 [1907.02454]
-
[41]
C. Dvorkin, T. Lin and K. Schutz,Cosmology of Sub-MeV Dark Matter Freeze-In,Phys. Rev. Lett.127(2021) 111301 [2011.08186]
-
[42]
Observational Constraints on Decoupled Hidden Sectors
M. Heikinheimo, T. Tenkanen, K. Tuominen and V. Vaskonen,Observational Constraints on Decoupled Hidden Sectors,Phys. Rev. D94(2016) 063506 [1604.02401]
work page internal anchor Pith review Pith/arXiv arXiv 2016
-
[43]
N. Bellomo, K.V. Berghaus and K.K. Boddy,Impact of freeze-in on dark matter isocurvature, JCAP11(2023) 024 [2210.15691]
- [44]
-
[45]
Yoon,Boltzmann Equation Solver for Thermalization,2603.28848
J.-H. Yoon,Boltzmann Equation Solver for Thermalization,2603.28848
-
[46]
L. Storchi, P. Campeti, M. Lattanzi, N. Antonini, E. Calore and P. Lubrano,gCAMB: A GPU-accelerated Boltzmann solver for next-generation cosmological surveys,Astron. Comput. 55(2026) 101038 [2509.25110]
- [47]
- [48]
-
[49]
S. Bhattacharya, S. Das, K. Dutta, M.R. Gangopadhyay, R. Mahanta and A. Maharana, Nonthermal hot dark matter from inflaton or moduli decay: Momentum distribution and relaxation of the cosmological mass bound,Phys. Rev. D103(2021) 063503 [2009.05987]
-
[50]
A. Ghosh and S. Mukhopadhyay,Momentum distribution of dark matter produced in inflaton decay: Effect of inflaton mediated scatterings,Phys. Rev. D106(2022) 043519 [2205.03440]
-
[51]
T. Moroi and W. Yin,Light Dark Matter from Inflaton Decay,JHEP03(2021) 301 [2011.09475]
-
[52]
G. Ballesteros, M.A.G. Garcia and M. Pierre,How warm are non-thermal relics? Lyman-α bounds on out-of-equilibrium dark matter,JCAP03(2021) 101 [2011.13458]
-
[53]
L. Johns and G.M. Fuller,Self-interacting sterile neutrino dark matter: the heavy-mediator case,Phys. Rev. D100(2019) 023533 [1903.08296]
-
[54]
M.D. Astros and S. Vogl,Boosting the production of sterile neutrino dark matter with self-interactions,JHEP03(2024) 032 [2307.15565]
-
[55]
M. Badziak and M. Laletin,Precise predictions for the QCD axion contribution to dark radiation with full phase-space evolution,JHEP02(2025) 108 [2410.18186]. – 24 –
- [56]
-
[57]
K. Ala-Mattinen, M. Heikinheimo, K. Kainulainen and K. Tuominen,Momentum distributions of cosmic relics: Improved analysis,Phys. Rev. D105(2022) 123005 [2201.06456]
-
[58]
A. Aboubrahim, M. Klasen and L.P. Wiggering,Forbidden dark matter annihilation into leptons with full collision terms,JCAP08(2023) 075 [2306.07753]
-
[59]
Taitano, L
W.T. Taitano, L. Chac´ on, A.N. Simakov and K. Molvig,A mass, momentum, and energy conserving, fully implicit, scalable algorithm for the multi-dimensional, multi-species Rosenbluth-Fokker-Planck equation,Journal of Computational Physics297(2015) 357. – 25 –
2015
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.