- [P] Equinox (1.3k stars), a JAX library for neural networks and sciML https://github.com/patrick-kidger/equinox 24 comments machinelearning
Linking pages
- JAX vs Julia (vs PyTorch) · Patrick Kidger https://kidger.site/thoughts/jax-vs-julia/ 110 comments
- Just know stuff. (Or, how to achieve success in a machine learning PhD.) · Patrick Kidger https://kidger.site/thoughts/just-know-stuff/ 103 comments
- [AINews] Google I/O in 60 seconds • Buttondown https://buttondown.email/ainews/archive/ainews-google-io-in-60-seconds/ 3 comments
- GitHub - patrick-kidger/optimistix: Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/ https://github.com/patrick-kidger/optimistix 1 comment
- GitHub - ml-tooling/best-of-ml-python: 🏆 A ranked list of awesome machine learning Python libraries. Updated weekly. https://github.com/ml-tooling/best-of-ml-python 0 comments
- GitHub - NTT123/pax: A stateful pytree library for training neural networks. https://github.com/NTT123/pax 0 comments
- Tutorial: Writing JAX-like code in PyTorch with functorch - Simone Scardapane https://www.sscardapane.it/tutorials/functorch/ 0 comments
- GitHub - google/jaxtyping: Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. https://github.com/google/jaxtyping 0 comments
- GitHub - stanford-crfm/haliax: Named Tensors for Legible Deep Learning in JAX https://github.com/stanford-crfm/haliax 0 comments
- GitHub - stanford-crfm/levanter: Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax https://github.com/stanford-crfm/levanter 0 comments
Linked pages
- GitHub - google/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more https://github.com/google/jax 99 comments
- [2111.00254] Equinox: neural networks in JAX via callable PyTrees and filtered transformations https://arxiv.org/abs/2111.00254 0 comments
- GitHub - google/flax: Flax is a neural network library for JAX that is designed for flexibility. https://github.com/google/flax/ 0 comments
- GitHub - deepmind/dm-haiku: JAX-based neural network library https://github.com/deepmind/haiku 0 comments
- GitHub - google/jaxtyping: Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. https://github.com/google/jaxtyping 0 comments
Would you like to stay up to date with Computer science? Checkout Computer science
Weekly.
Related searches:
Search whole site: site:github.com
Search title: GitHub - patrick-kidger/equinox: Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/
See how to search.