Reverse Dependencies of jax

The following projects have a declared dependency on jax:

• Equimo — Implementation of popular vision models in Jax
• equinox — Elegant easy-to-use neural networks in JAX.
• esch — esch (v.) : to turn matricies into high quality svg (animations)
• esm2quinox — ESM2 implemented in Equinox+JAX.
• esquilax — JAX multi-agent simulation and ML toolset
• essos — ESSOS: e-Stellarator Simulation and Optimization Suite
• etils — Collection of common python utils
• etna — ETNA is the first python open source framework of T-Bank AI Center. It is designed to make working with time series simple, productive, and fun.
• eulerpi — The eulerian parameter inference (eulerpi) returns a parameter distribution, which is consistent with the observed data by solving the inverse problem directly. In the case of a one-to-one mapping, this is the true underlying distribution.
• evaluators — Various scene understanding and perception evaluation metrics.
• evermore — Differentiable (binned) likelihoods in JAX.
• evo-science — Awesome evo_science created by maycuatroi
• evofr — Tools for evolutionary forecasting.
• evojax — EvoJAX: Hardware-accelerated Neuroevolution.
• evosax — Evolution Strategies in JAX
• ex2mcmc — Local-Global MCMC kernels: the bost of both worlds (NeurIPS 2022)
• exciting-environments — Physical differential equations wrapped into Gymnasium environments
• exojax — Auto-differentiable line-by-line spectral modeling of exoplanets/brown dwarfs using JAX.
• exoplanet-core — The compiled backend for exoplanet
• explainax — JAX-based Model Explanation and Interpretation Library
• exponax — Efficient differentiable PDE solvers in JAX.
• factopy — no summary
• fastax — A Jax based neural network library for research
• fastbrain — Python library of of jax-based representational studies in neuroscience and deep learning
• fastdla — Fast calculation of dynamical Lie algebra
• fastestimator — Deep learning framework
• fastfem — A Python package for solving PDEs with the finite element method and automatic differentiation
• fastreg — Fast sparse regressions
• fasttracks — A massively-parallel engine for generic CW detection statistics.
• FContin — Numerical continuation using just the function
• fdtdx — A JAX framework for the finite-difference time-domain (FDTD) method for solving Maxwell's equations with a focus on inverse design of photonic devices.
• fedjax — Federated learning simulation with JAX.
• fedml-afaf — A research and production integrated edge-cloud library for federated/distributed machine learning at anywhere at any scale.
• feedbax — Optimal feedback control + interventions in JAX.
• felafax — felafax
• felupe — Finite Element Analysis
• femr-cuda — Framework for Electronic Medical Records. A python package for building models using EHR data.
• femr-oldcpu — Framework for Electronic Medical Records. A python package for building models using EHR data.
• femtoscope — Python code for solving nonlinear PDEs from modified gravity models on unbounded domains
• fenbux — A Simple Statistical Distribution Library In JAX
• fennel-seed — Light-yields for tracks, and cascades
• fibermc — A Jax-based differentiable Monte Carlo estimator with applications to differentiable simulation, computational geometry, and topology optimization.
• fiction — Astronomical image processing with jax
• fiestaEM — Fast inference of electromagnetic signals with JAX
• FiniteDiffX — Finite difference tools in JAX
• fitree — no summary
• FixedPointJAX — Fixed-point iterations for root finding implemented in JAX
• fjformer — Embark on a journey of paralleled/unparalleled computational prowess with FJFormer in JAX
• flaim — Flax Image Models
• flarejax — Flexible Neural Networks in JAX
• flash-attention-jax — Flash Attention - in Jax
• flashbax — Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly with JAX's Just-In-Time (JIT) compilation.
• flax — Flax: A neural network library for JAX designed for flexibility
• flax-addons — flax addons
• flax-gated-linear-rnn — GatedLinearRNN Model
• flax-pilot — A Simplistic trainer for Flax
• flax-trainer — Flax Trainer
• flax-vision-models — A repository of Deep Learning models in Flax
• FlaxDiff — A versatile and easy to understand Diffusion library
• flaxmodels — A collection of pretrained models in Flax.
• flaxsr — Super Resolution tools with Jax/Flax
• FlexFL — Flexible Federated Learning Framework
• flora-opt — no summary
• floral — the best neural network library
• flowfunnel — Leveraging Bayesian hierarchical models to diagnose issues and proactively predict user flow stages in conversion funnels.
• flowjax — Easy to use distributions, bijections and normalizing flows in JAX.
• flowMC — Normalizing flow exhanced sampler in jax
• flt — Fast Legendre transform for NumPy and JAX
• fmmax — Fourier modal method with Jax
• folding — ColabFold modified for AlphaFold2 and local installation
• folx — Forward Laplacian for JAX
• foragax — ABM package in JAX.
• forecat — Keras based Forescast model builder
• fouriax — A jax port of auraloss
• fridom — Framework for Idealized Ocean Models
• fttjax — Feature Tokenizer + Transformer - JAX
• funprogram — A functional programming framework for python.
• funsor — A tensor-like library for functions and distributions
• galax — Galactic Dynamics in Jax.
• gallifrey — A JAX-based gaussian process structure learning package for time series modelling.
• galtab — Galaxy tabulation: HOD counts-in-cells statistics with JAX
• gamerl.dqn.jax — JAX implementation of Deep Q-Network
• gamerl.ppo.jax — JAX implementation of Proximal Policy Optimization
• gatv2-eqx — Add your description here
• gaul — no summary
• gdec — Linear decoders for angled grating stimuli
• gdsfactoryplus — GDSFactory+: adds powerful features such as foundry PDKs, simulations, and verification tools like DRC and LVS.
• gemma — Gemma open-weight LLM library from Google DeepMind.
• gemseo-jax — GEMSEO plugin for JAX (jit compilation, autodiff, XLA)
• generax — Generative Models using Jax
• genjax — Probabilistic programming with Gen, built on top of JAX.
• geodesix — A jax pytree library for spatial vector algebra
• geometricalgebra — A package for conformal geometric algebra
• geometricconvolutions — Package for building Convolutional Neural Networks on images of tensors.
• geometricjax-experimental — basic tests for full geometricjax library
• gfloat — Generic floating point handling in Python
• ghq — Gauss-Hermite quadrature in JAX
• gibss — Add your description here
• gigalens — Fast strong gravitational lens modeling
• ginjax — Package for building Convolutional Neural Networks on images of tensors.