diffprog / codeLinks
Code for the book "The Elements of Differentiable Programming".
☆255Updated last month
Alternatives and similar repositories for code
Users that are interested in code are comparing it to the libraries listed below
Sorting:
- Patched Attention for Nonlinear Dynamics☆156Updated this week
- Deep Learning, an Energy Approach☆199Updated last month
- Compositional Linear Algebra☆487Updated this week
- A projection-based framework for gradient-free and parallel learning☆103Updated last month
- Uncertainty quantification with PyTorch☆367Updated 3 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆450Updated 3 weeks ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆154Updated 9 months ago
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆210Updated last year
- Interactive textbook on state-space models☆196Updated last year
- Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach☆117Updated last year
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆276Updated 2 weeks ago
- Run PyTorch in JAX. 🤝☆266Updated 3 weeks ago
- Machine Learning with Symbolic Tensors☆324Updated 2 months ago
- ☆30Updated last week
- Notebooks for the "Deep Learning with JAX" book☆152Updated 2 months ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆445Updated last week
- Agustinus' very opiniated publication-ready plotting library☆67Updated 3 months ago
- ☆201Updated last week
- Multiple dispatch over abstract array types in JAX.☆127Updated last month
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆461Updated 2 weeks ago
- The boundary of neural network trainability is fractal☆215Updated last year
- JAX Arrays for human consumption☆106Updated last month
- Graph neural networks in JAX.☆67Updated last year
- Named Tensors for Legible Deep Learning in JAX☆197Updated this week
- ☆141Updated this week
- Bayesian inference with Python and Jax.☆34Updated 2 years ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆290Updated 11 months ago
- 🧱 Modula software package☆216Updated last week
- ☆144Updated 2 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆112Updated 4 months ago