Reverse Dependencies of jax

The following projects have a declared dependency on jax:

• simsopt — Framework for optimizing stellarators
• SiSo — no summary
• sklearn-jax-kernels — Composable kernels for scikit-learn implemented in JAX.
• skrl — Modular and flexible library for reinforcement learning on PyTorch and JAX
• skscope — Sparsity-Constraint OPtimization via itErative-algorithm
• slax — An SNN/RNN training package in JAX
• smeagol-gw — Companion package to Golum package for selection effects
• snaxlib — A simple deep learning library for JAX.
• softclip — JAX/Flax implementation of softclip
• somap — Self-Organizing Maps in python
• souJpg-diffusers — State-of-the-art diffusion in PyTorch and JAX.
• sourceinversion — A Gaussian plume model simulator and gas emission source inversion package.
• spaceTree — PyPI package for multi-task label transfer from single-cell refrence data to spatial data
• spanking — 🍑👋
• sparsejac — Efficient forward- and reverse-mode sparse Jacobians using Jax.
• sparsenn — no summary
• spbessax — spherical Bessel functions for JAX
• SPEADI — A Python package that aims to characterise the dynamics of local chemical environments
• specsy — Model fitting package for the chemical analysis of astronomical spectra
• specterpy — framework for defining, building, and evaluating generalized shape observables for collider physics
• speech-intelligibility-index — Speech Intelligibility Index
• spey-pyhf — pyhf plugin for spey interface
• sphericart-jax — JAX bindings to sphericart
• spikewrap — Run extracellular electrophysiology analysis with SpikeInterface
• spinn — Simple Physics Informed Neural Networks
• splinaltap — Keyframe interpolation and expression evaluation that goes to eleven!
• split-normal — A tiny package implementing functions of the split normal distribution compatible with Numpy and JAX.
• splotch-st — Splotch is a hierarchical generative probabilistic model for analyzing Spatial Transcriptomics (ST) data
• SpongeExt — Sponge Extension For Enhanced Sampling
• SpotDIPy — An Easy Way for Stellar Doppler Imaging of Cool Single Stars
• spotter — Approximate forward models of fluxes and spectra time-series of non-uniform stars.
• squishyplanet — A package for dealing with non-spherical exoplanets
• srb — Space Robotics Bench
• stablehlo-coreml-experimental — Convert StableHLO models into Apple Core ML format
• stadion — Causal Modeling with Stationary Diffusions
• stamox — Accelerate Your Statistical Analysis with JAX.
• starccato_jax — JAX-VAE stellar core collapse waveform
• starccato-sampler — Tools for parameter estimation with JAX-VAE model.
• starred-astro — A two-channel deconvolution method with Starlet regularization
• stars-omics — A spatial transcriptomics analysis tool.
• starspotter — Stellar contamination estimates from rotational light curves
• statedict2pytree — Converts torch models into PyTrees for Equinox
• statinf — A library for statistics and causal inference
• statistika — The ultimate statistics package in Python.
• statjax — Data science tools built with a Jax backend.
• stellar-spice — no summary
• stljax — stlcg with jax
• stories-jax — Learn spatially informed Waddington-like potentials for single-cell gene expression
• stormi — Probabilistic modeling of cellular dynamics
• strange — Online time series analysis
• stratmc — Bayesian statistical framework for reconstructing past Earth system change from the stratigraphic record
• streamfitter — A program for fitting functions and calculating montecarlo based errors
• summerepi2 — Summer is a compartmental disease modelling framework, written in Python. It provides a high-level API to build and run models.
• supergrad — SuperGrad differentiable Hamiltonian simulator
• supply-chains — Add your description here
• surjectors — Surjection layers for density estimation with normalizing flows
• swc-ephys — Manage extracellular electrophysiology analysis
• symjax — A Symbolic JAX software
• sympy2jax — Turn SymPy expressions into trainable JAX expressions.
• synecdoche — Synecdoche: Hypernetworks for Haiku in JAX
• synthax — SynthAX: A Fast Modular Synthesizer in JAX
• synthid-text — SynthID Text: Identifying AI-generated text content
• syrkis — no summary
• tables-io — Input/output and conversion functions for tabular data
• tactile-mnist — The Tactile MNIST benchmark for active tactile perception.
• tax-dp — A fully sharded data parallel trainer for jax
• taxim — GPU accelerated implementation of the Taxim simulator.
• tcup — no summary
• tdastro — no summary
• tdsxtract — Tools for extracting permittivity values from Terahertz Time Domain Spectroscopy measurements.
• tencirchem-ng — Efficient quantum computational chemistry based on TensorCircuit
• teneva-ht-jax — Compact implementation of basic operations in the Hierarchical Tucker (HT) format for approximation and sampling from multidimensional arrays and multivariate functions
• teneva-jax — Compact implementation of basic operations in the tensor-train (TT) format with jax framework for approximation, optimization and sampling with multidimensional arrays and multivariate functions
• tensorcircuit — High performance unified quantum computing framework for the NISQ era
• tensorcircuit-ng — High performance unified quantum computing framework for the NISQ era
• tensorcircuit-nightly — nightly release for tensorcircuit
• tensorflow-datasets — tensorflow/datasets is a library of datasets ready to use with TensorFlow.
• tensorflow-federated — TensorFlow Federated is an open-source federated learning framework.
• tensorflow-ml — Tensorflow ML
• tensorflow-probability — Probabilistic modeling and statistical inference in TensorFlow
• tensorflowjs — no summary
• tensorshare — 🤝 Trade any tensors over the network
• tensorwaves — Python fitter package for multiple computational back-ends
• tensorwrap — TensorWrap: A high level TensorFlow wrapper for JAX.
• teos10 — Unofficial Python implementation of the TEOS-10 properties of water
• testax — Jit-able runtime assertions for jax in numpy style.
• tetris-gymnasium — A fully configurable Gymnasium compatible Tetris environment
• tf-shb-gabriel-0302 — State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch
• tf2jax — TF2JAX: Convert TensorFlow to JAX
• tfc — Theory of Functional Connections (TFC): A functional interpolation framework with applications in differential equations.
• tfds-nightly — tensorflow/datasets is a library of datasets ready to use with TensorFlow.
• tfp-nightly — Probabilistic modeling and statistical inference in TensorFlow
• tfx — TensorFlow Extended (TFX) is a TensorFlow-based general-purpose machine learning platform implemented at Google.
• theoval — Theoval
• thermoextrap — Thermodynamic extrapolation
• thermox — Exact OU processes with JAX
• thunderbeard — Minimal python deep learning library
• ticktack — Python package to build a carbon box model.
• tidy3d — A fast FDTD solver
• tidy3d-beta — A fast FDTD solver