pith. sign in

arxiv: 2607.00755 · v1 · pith:FWPCVTJ5new · submitted 2026-07-01 · ✦ hep-ph

KineticXGPU: A Tensorized Collision Operator for Dark-Sector Self-Scattering

Pith reviewed 2026-07-02 10:24 UTC · model grok-4.3

classification ✦ hep-ph
keywords dark sectorself-scatteringcollision operatorGPU computingfreeze-inmomentum distributionMaxwell-Boltzmann
0
0 comments X

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.

The paper presents KineticXGPU, a PyTorch code that rewrites the discretized 2-to-2 elastic self-collision operator for dark-sector momentum distributions as tensor contractions. This formulation runs efficiently on GPUs and is applied to a two-source freeze-in process that produces a bimodal final distribution. The calculation demonstrates that stronger elastic self-interactions progressively remove the bimodality and drive the distribution toward a Maxwell-Boltzmann shape. The work also compares the full phase-space evolution with a set of fluid equations for number density and velocity dispersion and reports faster runtimes on GPU than on CPU.

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

These are editorial extensions of the paper, not claims the author makes directly.

  • 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.

Desk editor's note, referee report, simulated authors' rebuttal, and a circularity audit. Tearing a paper down is the easy half of reading it; the pith above is the substance, this is the friction.

Referee Report

1 major / 0 minor

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)
  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

1 responses · 0 unresolved

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
  1. 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

0 steps flagged

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

0 free parameters · 0 axioms · 0 invented entities

Only the abstract is available; no explicit free parameters, axioms, or invented entities are identifiable from the provided text.

pith-pipeline@v0.9.1-grok · 5649 in / 1296 out tokens · 37676 ms · 2026-07-02T10:24:54.308617+00:00 · methodology

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Reference graph

Works this paper leans on

59 extracted references · 56 canonical work pages · 19 internal anchors

  1. [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. [2]

    Dark Matter

    M. Cirelli, A. Strumia and J. Zupan,Dark Matter,2406.01705

  3. [3]

    Arcadi, D

    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. [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]

  5. [5]

    J.L. Feng, M. Kaplinghat and H.-B. Yu,Sommerfeld Enhancements for Thermal Relic Dark Matter,Phys. Rev. D82(2010) 083525 [1005.4678]

  6. [6]

    Binder, T

    T. Binder, T. Bringmann, M. Gustafsson and A. Hryczuk,Early kinetic decoupling of dark matter: when the standard way of calculating the thermal relic density fails,Phys. Rev. D96 (2017) 115010 [1706.07433]

  7. [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]

  8. [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

  9. [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]

  10. [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]

  11. [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]

  12. [12]

    Carlson, M.E

    E.D. Carlson, M.E. Machacek and L.J. Hall,Self-interacting dark matter,Astrophys. J.398 (1992) 43

  13. [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 –

  14. [14]

    The SIMP Miracle

    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]

  15. [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]

  16. [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

  17. [17]

    Erickcek, P

    A.L. Erickcek, P. Ralegankar and J. Shelton,Cannibal domination and the matter power spectrum,Phys. Rev. D103(2021) 103508 [2008.04311]

  18. [18]

    Heimersheim, N

    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. [19]

    Hufnagel and M.H.G

    M. Hufnagel and M.H.G. Tytgat,The domain of a cannibal dark matter,JCAP09(2023) 012 [2212.09759]

  20. [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]

  21. [21]

    March-Russell, H

    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. [22]

    $Z_2$ SIMP Dark Matter

    N. Bernal and X. Chu,Z 2 SIMP Dark Matter,JCAP01(2016) 006 [1510.08527]

  23. [23]

    Cervantes and A

    E. Cervantes and A. Hryczuk,Freezing-in cannibal dark sectors,JHEP11(2024) 050 [2407.12104]

  24. [24]

    Bernal, E

    N. Bernal, E. Cervantes, K. Deka and A. Hryczuk,Freezing-in cannibals with low-reheating temperature,JHEP09(2025) 083 [2506.09155]

  25. [25]

    Constraints on the Self-Interaction Cross-Section of Dark Matter from Numerical Simulations of the Merging Galaxy Cluster 1E 0657-5

    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]

  26. [26]

    Hryczuk and M

    A. Hryczuk and M. Laletin,Impact of dark matter self-scattering on its relic abundance,Phys. Rev. D106(2022) 023007 [2204.07078]

  27. [27]

    Chatterjee and A

    S. Chatterjee and A. Hryczuk,Conversions in two-component dark sectors: a phase space level analysis,JHEP07(2025) 279 [2502.08725]

  28. [28]

    Exploring non-equilibrium effects in sequential freeze-in

    S. Chatterjee and A. Hryczuk,Exploring non-equilibrium effects in sequential freeze-in, 2604.14688

  29. [29]

    Y. Du, F. Huang, H.-L. Li, Y.-Z. Li and J.-H. Yu,Revisiting dark matter freeze-in and freeze-out through phase-space distribution,JCAP04(2022) 012 [2111.01267]

  30. [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]

  31. [31]

    D’Eramo and A

    F. D’Eramo and A. Lenoci,Lower mass bounds on FIMP dark matter produced via freeze-in, JCAP10(2021) 045 [2012.01446]

  32. [32]

    Dienes, F

    K.R. Dienes, F. Huang, J. Kost, S. Su and B. Thomas,Deciphering the archaeological record: Cosmological imprints of nonminimal dark sectors,Phys. Rev. D101(2020) 123511 [2001.02193]

  33. [33]

    Decant, J

    Q. Decant, J. Heisig, D.C. Hooper and L. Lopez-Honorez,Lyman-αconstraints on freeze-in and superWIMPs,JCAP03(2022) 041 [2111.09321]

  34. [34]

    Dienes, F

    K.R. Dienes, F. Huang, J. Kost, B. Thomas and H.-B. Yu,Evaluating Lyman-αconstraints for general dark-matter velocity distributions: Multiple scales and cautionary tales,Phys. Rev. D 106(2022) 123521 [2112.09105]. – 23 –

  35. [35]

    D’Eramo, A

    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. [36]

    Zhao, Y.-C

    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. [37]

    Machine Learning Does It and Does It Better: Unearthing Primordial Dark-Matter Velocities from the Matter Power Spectrum

    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

  38. [38]

    Thoss, L

    V. Thoss, L. Lopez-Honorez, F. K¨ uhnel and M. Hufnagel,Non-Cold Dark Matter from Memory-Burdened Primordial Black Holes,2604.00090

  39. [39]

    The Evolution of the Intergalactic Medium

    M. McQuinn,The Evolution of the Intergalactic Medium,Ann. Rev. Astron. Astrophys.54 (2016) 313 [1512.00086]

  40. [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. [41]

    Dvorkin, T

    C. Dvorkin, T. Lin and K. Schutz,Cosmology of Sub-MeV Dark Matter Freeze-In,Phys. Rev. Lett.127(2021) 111301 [2011.08186]

  42. [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]

  43. [43]

    Bellomo, K.V

    N. Bellomo, K.V. Berghaus and K.K. Boddy,Impact of freeze-in on dark matter isocurvature, JCAP11(2023) 024 [2210.15691]

  44. [44]

    Binder, T

    T. Binder, T. Bringmann, M. Gustafsson and A. Hryczuk,DRAKE: Dark matter Relic Abundance beyond Kinetic Equilibrium,Eur. Phys. J. C81(2021) 577 [2103.01944]

  45. [45]

    Yoon,Boltzmann Equation Solver for Thermalization,2603.28848

    J.-H. Yoon,Boltzmann Equation Solver for Thermalization,2603.28848

  46. [46]

    Storchi, P

    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. [47]

    O. Hahn, F. List and N. Porqueres,DISCO-DJ I: a differentiable Einstein-Boltzmann solver for cosmology,JCAP06(2024) 063 [2311.03291]

  48. [48]

    Z. Zhou, C. Giovanetti and H. Liu,ABCMB: A Python+JAX Package for the Cosmic Microwave Background Power Spectrum,2602.15104

  49. [49]

    Bhattacharya, S

    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. [50]

    Ghosh and S

    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. [51]

    Moroi and W

    T. Moroi and W. Yin,Light Dark Matter from Inflaton Decay,JHEP03(2021) 301 [2011.09475]

  52. [52]

    Ballesteros, M

    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. [53]

    Johns and G.M

    L. Johns and G.M. Fuller,Self-interacting sterile neutrino dark matter: the heavy-mediator case,Phys. Rev. D100(2019) 023533 [1903.08296]

  54. [54]

    Astros and S

    M.D. Astros and S. Vogl,Boosting the production of sterile neutrino dark matter with self-interactions,JHEP03(2024) 032 [2307.15565]

  55. [55]

    Badziak and M

    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. [56]

    M. Jain, A. Maggi, W.-Y. Ai and D.J.E. Marsh,New insights into axion freeze-in,JHEP11 (2024) 166 [2406.01678]. [61]Planckcollaboration,Planck 2018 results. VI. Cosmological parameters,Astron. Astrophys. 641(2020) A6 [1807.06209]

  57. [57]

    Ala-Mattinen, M

    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. [58]

    Aboubrahim, M

    A. Aboubrahim, M. Klasen and L.P. Wiggering,Forbidden dark matter annihilation into leptons with full collision terms,JCAP08(2023) 075 [2306.07753]

  59. [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 –