JOSS Papers: JAX

All Papers
3111
 Published Papers
2783
 Active Papers
328
published Published 21 days ago

ReLax: Efficient and Scalable Recourse Explanation Benchmarking using JAX

Python Jupyter Notebook SCSS
@BirkhoffG
10.21105/joss.06567
published Published 3 months ago

squishyplanet: modeling transits of non-spherical exoplanets in JAX

Python Jupyter Notebook
@ben-cassese
10.21105/joss.06972
published Published 4 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 5 months ago

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

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

JAXbind: Bind any function to JAX

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

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

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

Surjectors: surjection layers for density estimation with normalizing flows

Python
@dirmeier
10.21105/joss.06188
published Published about 1 year ago

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

@prappleizer
10.21105/joss.05703
published Published about 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
Displaying papers 1 - 10 of 19 in total

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

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