JOSS Papers: JAX

All Papers
2968
 Published Papers
2651
 Active Papers
317
published Published 13 days ago

squishyplanet: modeling transits of non-spherical exoplanets in JAX

Python Jupyter Notebook
@ben-cassese
10.21105/joss.06972
published Published about 2 months ago

IKPLS: Improved Kernel Partial Least Squares and Fast Cross-Validation Algorithms for Python with CPU and GPU Implementations Using NumPy and JAX

Python
@Sm00thix
10.21105/joss.06533
published Published 2 months ago

Catalyst: a Python JIT compiler for auto-differentiable hybrid quantum programs

Python
@josh146
10.21105/joss.06720
published Published 3 months ago

JAXbind: Bind any function to JAX

Python C++
@roth-jakob
10.21105/joss.06532
published Published 6 months ago

Φ-ML: Intuitive Scientific Computing with Dimension Types for Jax, PyTorch, TensorFlow & NumPy

Python C++
@holl-
10.21105/joss.06171
published Published 7 months ago

Surjectors: surjection layers for density estimation with normalizing flows

Python
@dirmeier
10.21105/joss.06188
published Published 12 months ago

pysersic: A Python package for determining galaxy structural properties via Bayesian inference, accelerated with jax

@prappleizer
10.21105/joss.05703
published Published almost 1 year ago

qujax: Simulating quantum circuits with JAX

Python
@SamDuffield
10.21105/joss.05504
published Published over 1 year ago

flowMC: Normalizing flow enhanced sampling package for probabilistic inference in JAX

Python
@kazewong
10.21105/joss.05021
published Published over 1 year ago

PyKronecker: A Python Library for the Efficient Manipulation of Kronecker Products and Related Structures

Python
@nickelnine37
10.21105/joss.04900
Displaying papers 1 - 10 of 18 in total

Journal of Open Source Software is an affiliate of the Open Source Inititative.

Journal of Open Source Software is part of Open Journals, which is a NumFOCUS-sponsored project.

Table of Contents
Public user content licensed CC BY 4.0 unless otherwise specified.
ISSN 2475-9066