Reverse Dependencies of jax
The following projects have a declared dependency on jax:
- sbmltoodejax — lightweight library that allows to automatically parse and convert SBML models into python models written end-to-end in JAX
- sbx-rl — Jax version of Stable Baselines, implementations of reinforcement learning algorithms.
- scandeval — Evaluation of pretrained language models on mono- or multilingual language tasks.
- scarlet2 — Scarlet, all new and shiny
- scatrex — Map single-cell transcriptomes to copy number evolutionary trees.
- scdef — Extract hierarchical signatures of cell state from single-cell data.
- scenario-generator — Scenario generator for understanding Filecoin dynamics
- scib-metrics — Accelerated and Python-only scIB metrics
- scKinetics — Biological prior guided single-cell kinetics inference.
- scvi-tools — Deep probabilistic analysis of single-cell omics data.
- searchep — A module to search for exceptional points
- secretflow — SecretFlow
- secretflow-lite — SecretFlow Lite
- segnn-jax — Steerable E(3) GNN in jax
- Sella — no summary
- sepes — Visualize, create, and operate on nested trees in the most intuitive way possible.
- seqio — SeqIO: Task-based datasets, preprocessing, and evaluation for sequence models.
- seqio-nightly — SeqIO: Task-based datasets, preprocessing, and evaluation for sequence models.
- sequel-core — A Continual Learning Framework for both Jax and PyTorch.
- sequential-ft-transformer — FT Transformer applied to sequential tabular data
- serket — Finite difference tools in JAX
- SFcalculator-jax — A Differentiable pipeline connecting molecule models and crystallpgraphy data
- sgGWR — no summary
- SGMCMCJax — SGMCMC samplers in JAX
- shamnet — Differentiable abundance matching
- shbtf0302 — State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch
- shml — lightweight pdata cleaning/processing/plotting/ML training library for use with an ATLAS BSM dihiggs search
- sibylla — A Jax package for Gradient Descent Image Reconstruction
- sidious — no summary
- sigkerax — Signature kernel computations via PDE solvers in JAX
- signax — Differentiable signature calculations in JAX.
- simphony — Simphony: A simulator for photonic circuits
- simphox — no summary
- simple-pytree — no summary
- 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.
- spanking — 🍑👋
- sparsejac — Efficient forward- and reverse-mode sparse Jacobians using Jax.
- sparsenn — no summary
- 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
- speech-intelligibility-index — Speech Intelligibility Index
- spey-pyhf — pyhf plugin for spey interface
- spikewrap — Run extracellular electrophysiology analysis with SpikeInterface
- spinn — Simple Physics Informed Neural Networks
- split-normal — A tiny package implementing functions of the split normal distribution compatible with Numpy and JAX.
- squishyplanet — A package for dealing with non-spherical exoplanets
- stadion — Causal Modeling with Stationary Diffusions
- stamox — Accelerate Your Statistical Analysis with JAX.
- 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.
- stellar-spice — no summary
- strange — Online time series analysis
- streamfitter — Add your description here
- summerepi2 — Summer is a compartmental disease modelling framework, written in Python. It provides a high-level API to build and run models.
- surjectors — Surjection layers for density estimation with normalizing flows
- swc-ephys — Manage extracellular electrophysiology analysis
- syft — Perform numpy-like analysis on data that remains in someone elses server
- 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
- tables-io — Input/output and conversion functions for tabular data
- tdsxtract — Tools for extracting permittivity values from Terahertz Time Domain Spectroscopy measurements.
- 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-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-federated-nightly — 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.
- testax — Jit-able runtime assertions for jax in numpy style.
- 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
- thunderbeard — Minimal python deep learning library
- ticktack — Python package to build a carbon box model.
- tidy3d — A fast FDTD solver