Alternatives and similar repositories for jax-tutorial:

Users that are interested in jax-tutorial are comparing it to the libraries listed below

• pierreglaser / jupyter-slurm-setup-instructions
  ☆20Updated 8 months ago
• mackelab / neuralgbi
  Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inferenc…
  ☆10Updated last year
• mlcolab / sbi-workshop
  SBI Workshop jointly by Helmholtz AI + ML ⇌ Science Colaboratory
  ☆22Updated last year
• jaxleyverse / jaxley
  Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.
  ☆73Updated this week
• EEA-sensors / sqrt-parallel-smoothers
  A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…
  ☆12Updated last month
• sbi-dev / pyknos
  Conditional density estimation with neural networks
  ☆30Updated last week
• mackelab / simformer
  Combination of transformers and diffusion models for flexible all-in-one simulation-based inference
  ☆56Updated 7 months ago
• lindermanlab / ssm-jax
  Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend
  ☆58Updated 7 months ago
• berenslab / RFEst
  A Python 3 toolbox for neural receptive field estimation using splines and Gaussian priors.
  ☆24Updated 5 months ago
• dieterichlawson / snax
  ☆12Updated last year
• timkimd / plnde
  Code for "Inferring Latent Dynamics Underlying Neural Population Activity via Neural Differential Equations"
  ☆9Updated 3 years ago
• petergchang / sarkka-jax
  ☆11Updated 2 years ago
• mackelab / delfi
  Density estimation likelihood-free inference. No longer actively developed see https://github.com/mackelab/sbi instead
  ☆73Updated 4 years ago
• tachukao / mgplvm-pytorch
  ☆24Updated last year
• wesselb / lab
  A generic interface for linear algebra backends
  ☆71Updated 7 months ago
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆29Updated last week
• DurstewitzLab / GTF-shPLRNN
  ☆13Updated 9 months ago
• acerbilab / pybads
  PyBADS: Bayesian Adaptive Direct Search optimization algorithm for model fitting in Python
  ☆71Updated last week
• flatironinstitute / figurl
  Shareable, interactive scientific figures in the cloud
  ☆35Updated 2 months ago
• lindermanlab / PPSeq.jl
  Neyman-Scott point process model to identify sequential firing patterns in high-dimensional spike trains
  ☆63Updated last year
• anirgalgali / residual-dynamics
  Code for Galgali et al, 2023
  ☆12Updated 2 years ago
• catniplab / latent_dynamics_workshop
  Exercises and examples for the latent dynamics workshop
  ☆17Updated last year
• marineschimel / ilqr_vae
  ☆11Updated 2 years ago
• wesselb / gpar
  Implementation of the Gaussian Process Autoregressive Regression Model
  ☆62Updated last week
• INM-6 / nnmt
  Neuronal Network Meanfield Toolbox - official documentation: https://nnmt.readthedocs.io
  ☆13Updated last year
• flatironinstitute / spikeforest_old
  SpikeForest -- spike sorting analysis for website
  ☆16Updated 4 years ago
• JRBCH / spiffyplots
  A collection of matplotlib style sheets and plotting tools for publication-ready figures
  ☆14Updated 2 years ago
• lee-group-cmu / lf2i
  Likelihood-Free Frequentist Inference
  ☆15Updated last week
• brian-team / brian2modelfitting
  Model fitting toolbox for the Brian 2 simulator
  ☆14Updated last year
• bgroenks96 / SimulationBasedInference.jl
  A flexible toolkit for simulation based inference in Julia
  ☆18Updated last week