Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
awesome-equinox
Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox
https://github.com/matteoguarrera/awesome-equinox
- PaLM-jax - Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework) <img src="https://img.shields.io/github/stars/lucidrains/PaLM-jax?style=social" align="center">
- mistral_jax - This is a port of Mistral-7B model in JAX <img src="https://img.shields.io/github/stars/AakashKumarNain/mistral_jax?style=social" align="center">
- levanter - Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax <img src="https://img.shields.io/github/stars/stanford-crfm/levanter?style=social" align="center">
- eqxvision - A Python package of computer vision models for the Equinox ecosystem. <img src="https://img.shields.io/github/stars/paganpasta/eqxvision?style=social" align="center">
- haliax - Named Tensors for Legible Deep Learning in JAX <img src="https://img.shields.io/github/stars/stanford-crfm/haliax?style=social" align="center">
- diffrax - kidger/diffrax?style=social" align="center">
- lineax - Linear solvers in JAX and Equinox. <img src="https://img.shields.io/github/stars/patrick-kidger/lineax?style=social" align="center">
- optimistix - Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. <img src="https://img.shields.io/github/stars/patrick-kidger/optimistix?style=social" align="center">
- sympy2jax - Turn SymPy expressions into trainable JAX expressions. <img src="https://img.shields.io/github/stars/patrick-kidger/sympy2jax?style=social" align="center">
- flowMC - Normalizing-flow enhanced sampling package for probabilistic inference in Jax <img src="https://img.shields.io/github/stars/kazewong/flowMC?style=social" align="center">
- flowjax - FlowJax: Distributions and Normalizing Flows in Jax <img src="https://img.shields.io/github/stars/danielward27/flowjax?style=social" align="center">
- traceax - Traceax: Stochastic trace estimation using JAX <img src="https://img.shields.io/github/stars/mancusolab/traceax?style=social" align="center">
- galax - Galactic and Gravitational Dynamics in Python (+ GPU and autodiff) <img src="https://img.shields.io/github/stars/GalacticDynamics/galax?style=social" align="center">
- coordinax - Coordinates in JAX <img src="https://img.shields.io/github/stars/GalacticDynamics/coordinax?style=social" align="center">
- unxt - Unitful Quantities in JAX <img src="https://img.shields.io/github/stars/GalacticDynamics/unxt?style=social" align="center">
- statedict2pytree - Transforming a PyTorch model into an JAX PyTree <img src="https://img.shields.io/github/stars/Artur-Galstyan/statedict2pytree?style=social" align="center">
- jaxtyping
- Optax - order gradient (SGD, Adam, ...) optimisers.
- Orbax - host/multi-device).
- BlackJAX
- PySR - JAX honourable mention!)
- Awesome JAX
Keywords
jax
14
equinox
4
deep-learning
4
machine-learning
2
neural-networks
2
dynamical-systems
2
normalizing-flows
2
python
1
transformers
1
attention-mechanism
1
artificial-general-intelligence
1
units
1
quantities
1
array
1
coordinates
1
coordinate-transformation
1
coordinate-systems
1
stellar-dynamics
1
gpu-acceleration
1
galaxy
1
dynamics
1
auto-differentiation
1
astrophysics
1
astronomy
1
trace-est
1
statistics
1
python3
1
machine-learn
1
xla
1
numpy
1
neural-network
1
awesome-list
1
awesome
1
autograd
1
sampling-methods
1
probabilistic-programming
1
hamiltonian-monte-carlo
1
bayesian-inference
1
flax
1
checkpoint
1
typing
1
python-typing
1
normalising-flows
1
stat
1
mcmc
1
sympy
1
optimization
1
optimisation
1
linear-algebra
1
neural-differential-equations
1