Related papers: LINX: A Fast, Differentiable, and Extensible Big B…
We introduce Lineax, a library bringing linear solves and linear least-squares to the JAX+Equinox scientific computing ecosystem. Lineax uses general linear operators, and unifies linear solves and least-squares into a single,…
High-fidelity binaural audio synthesis is crucial for immersive listening, but existing methods require extensive computational resources, limiting their edge-device application. To address this, we propose the Lightweight Implicit Neural…
The abundances of light elements based on the big bang nucleosynthesis model are calculated using the Tsallis non-extensive statistics. The impact of the variation of the non-extensive parameter $q$ from the unity value is compared to…
Big-Bang Nucleosynthesis (BBN) predictions of primordial light-element abundances offer a powerful probe of early-Universe physics. However, high-accuracy numerical BBN calculations have become a major computational bottleneck for…
Large language models (LLMs) have demonstrated broad utility across molecular domains, spanning drug discovery and materials design. Analyzing LLMs' latent representations is crucial for elucidating their underlying mechanisms, improving…
The lattice Boltzmann method (LBM) has emerged as a prominent technique for solving fluid dynamics problems due to its algorithmic potential for computational scalability. We introduce XLB library, a Python-based differentiable LBM library…
Given the pressing need for assuring algorithmic transparency, Explainable AI (XAI) has emerged as one of the key areas of AI research. In this paper, we develop a novel Bayesian extension to the LIME framework, one of the most widely used…
We introduce JPC, a JAX library for training neural networks with Predictive Coding. JPC provides a simple, fast and flexible interface to train a variety of PC networks (PCNs) including discriminative, generative and hybrid models. Unlike…
For a brief time in its early evolution the Universe was a cosmic nuclear reactor. The expansion and cooling of the Universe limited this epoch to the first few minutes, allowing time for the synthesis in astrophysically interesting…
The concordance of standard big bang nucleosynthesis theory and the related observations of the light element isotopes (including some new higher \he4 abundances) will be reviewed. Implications of BBN on chemical evolution, dark matter and…
We examine the possibility that Big Bang Nucleosynthesis (BBN) may produce non-trivial amounts of $^6$Li. If a primordial component of this isotope could be observed, it would provide a new fundamental test of Big-Bang cosmology, as well as…
Big Bang Nucleosynthesis is the theory of the production of the the light element isotopes of D, He3, He4, and Li7. After a brief review of the essential elements of the standard Big Bang model at a temperature of about 1 MeV, the…
bde is a user-friendly Python package for Bayesian Deep Ensembles with a particular focus on tabular data. Built on an efficient JAX implementation of the sampling-based inference method Microcanonical Langevin Ensembles (MILE), it provides…
One of the three testaments in favor of the big bang theory is the prediction of the primordial elemental abundances in the big-bang nucleosynthesis (BBN). The Standard BBN is a parameter-free theory due to the precise knowledge of the…
The measurement of properties of exotic nuclei, essential for fundamental nuclear physics, now confronts a formidable challenge for contemporary radiofrequency accelerator technology. A promising option can be found in the combination of…
We introduce microJAX, the first fully differentiable implementation of the image-centered ray-shooting (ICRS) algorithm for gravitational microlensing. Built on JAX and its XLA just-in-time compiler, microJAX exploits GPU parallelism while…
We present CosmoPower-JAX, a JAX-based implementation of the CosmoPower framework, which accelerates cosmological inference by building neural emulators of cosmological power spectra. We show how, using the automatic differentiation, batch…
We present linrax, the first simplex based linear program (LP) solver compatible with the JAX ecosystem. In many control algorithms, LPs are often automatically generated and frequently solved either offline or online in the control loop.…
The importance of the Big Bang Nucleosynthesis (BBN) as a unique tool for studying neutrino properties is discussed, and the recent steps towards a self-consistent and robust handling of the weak reaction decoupling from the thermal bath as…
In the primordial Big Bang nucleosynthesis (BBN), only the lightest nuclides (D, $^3$He, $^4$He, and $^7$Li) were synthesized in appreciable quantities, and these relics provide us a unique window on the early universe. Currently, BBN…