37 projects
boundinstance
Weakref-based instance descriptors.
phasecurvefit
Walk through phase-space observations
unxts.linalg
Heterogeneous-unit matrices and vectors for unxt (canonical: unxts.linalg)
spexial
scipy.special in JAX.
xmmutablemap
Immutable Map, compatible with JAX & Equinox
unxt
Quantities in JAX
potamides
Constrain gravitational potential with stream curvature
diffraxtra
diffrax extras: OOP and vectorization
galax
Galactic Dynamics in Jax.
unxts.hypothesis
Hypothesis strategies for unxt (canonical: unxts.hypothesis)
unxts.api
Abstract dispatch API definitions for unxt (canonical: unxts.api)
optional-dependencies
Construct Checks for Optional Dependencies
quax
Multiple dispatch in JAX via custom interpreters.
quax-blocks
Building blocks for Quax classes
unxts.parametric
Dimension-parametrized quantities for unxt (canonical: unxts.parametric)
unxts.interop.matplotlib
matplotlib integration for unxt (canonical: unxts.interop.matplotlib)
unxts.interop.xarray
xarray integration for unxt (canonical: unxts.interop.xarray)
unxts.interop.gala
gala integration for unxt (canonical: unxts.interop.gala)
unxt-api
Abstract dispatch API definitions for unxt
unxt-hypothesis
Hypothesis strategies for unxt
jaxmore
There's more to JAX.
zeroth
Efficiently get the index-0 element of an iterable.
coordinax
Coordinates in JAX
quaxed
Pre-quaxed libraries for multiple dispatch over abstract array types in JAX
dataclassish
dataclass tools, extended by multiple dispatch
jax-bounded-while
A bounded (and autodiff friendly) while loop in JAX.
oncequinox
Create singleton Equinox modules.
cosmology.api
Cosmology API standard
mvgkde
MultiVariate Gaussian Kernel Density Estimator
plotting-backends
Plotting dispatch backends
is-annotated
Check if an object is an `Annotated` type.
jax-quantity
Quantities in JAX
array-api-jax-compat
Array-API JAX compatibility
overload-numpy
Overload NumPy Functions
cosmology
Python package for cosmology
macro-lightning
Constraining Macro Dark Matter Models with Lightning.
utilipy
Python grab-bag.
