zsteve's Stars
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
pyro-ppl/pyro
Deep universal probabilistic programming with Python and PyTorch
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
rtqichen/torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
pyvista/pyvista
3D plotting and mesh analysis through a streamlined interface for the Visualization Toolkit (VTK)
marcomusy/vedo
A python module for scientific analysis of 3D data based on VTK and Numpy
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
DiffEqML/torchdyn
A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
jakobrunge/tigramite
Tigramite is a python package for causal inference with a focus on time series data. The Tigramite documentation is at
atong01/conditional-flow-matching
TorchCFM: a Conditional Flow Matching library
dit/dit
Python package for information theory.
ott-jax/ott
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
Murali-group/Beeline
BEELINE: evaluation of algorithms for gene regulatory network inference
neurolabusc/nii2mesh
Convert NIfTI volume to triangulated mesh using marching cubes
HantaoShu/DeepSEM
JEFworks-Lab/STalign
Python tool for alignment of spatial transcriptomics (ST) data using diffeomorphic metric mapping
BirkhoffG/jax-dataloader
Pytorch-like dataloaders in JAX.
brianlee324/torch-lddmm
GPU and CPU implementation of LDDMM image registration
MellonNet/mln-backend-emulator
Server emulator for the My LEGO Network social network / game hybrid.
necludov/jam
Implementation of Action Matching
causalbench/causalbench
enhuiz/langevin-dynamics-and-score-matching
A demo shows how to combine Langevin dynamics with score matching for generative models.
mcbal/spin-model-transformers
Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
normal-computing/thermox
Exact OU processes with JAX
franciscovargas/GP_Sinkhorn
GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134
gioda/FeARLesS
4D reconstruction of developmental trajectories using spherical harmonics
ssp3nc3r/ggSankeyGrad
R package to create a ggplot-based Sankey graphic that encodes flows with a color gradient.
MatthewFlamm/pyransame
PYthon RAndom SAmpling for MEshes
alexisbellot/Graphical-modelling-continuous-time
Graphical modelling with time series data using an ODE model
agnesedaniele/GridPythonModule
A Python module to manipulate and simplify grid diagrams