28 projects
zeroth
Efficiently get the index-0 element of an iterable.
coordinax
Coordinates in JAX
unxt
Quantities in JAX
quax-blocks
Building blocks for Quax classes
quaxed
Pre-quaxed libraries for multiple dispatch over abstract array types in JAX
quax
Multiple dispatch in JAX via custom interpreters.
potamides
Constrain gravitational potential with stream curvature
jaxmore
There's more to JAX.
dataclassish
dataclass tools, extended by multiple dispatch
unxt-hypothesis
Hypothesis strategies for unxt
unxt-api
Abstract dispatch API definitions for unxt
jax-bounded-while
A bounded (and autodiff friendly) while loop in JAX.
phasecurvefit
Walk through phase-space observations
oncequinox
Create singleton Equinox modules.
xmmutablemap
Immutable Map, compatible with JAX & Equinox
optional-dependencies
Construct Checks for Optional Dependencies
diffraxtra
diffrax extras: OOP and vectorization
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.
galax
Galactic Dynamics in Jax.
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.