JOSS Papers: JAX

All Papers
3349
 Published Papers
2952
 Active Papers
397
published Published 9 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 10 months ago

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

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

JAXbind: Bind any function to JAX

Python C++
@roth-jakob
10.21105/joss.06532
published Published about 1 year ago

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

Python C++
@holl-
10.21105/joss.06171
published Published about 1 year ago

Surjectors: surjection layers for density estimation with normalizing flows

Python
@dirmeier
10.21105/joss.06188
published Published over 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 over 1 year ago

qujax: Simulating quantum circuits with JAX

Python
@SamDuffield
10.21105/joss.05504
published Published about 2 years ago

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

Python
@kazewong
10.21105/joss.05021
published Published about 2 years ago

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

Python
@nickelnine37
10.21105/joss.04900
published Published over 2 years ago

GPJax: A Gaussian Process Framework in JAX

Python
@thomaspinder
10.21105/joss.04455
Displaying papers 11 - 20 of 27 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