JAX implementation of Learning to learn by gradient descent by gradient descent
☆28Aug 5, 2025Updated 7 months ago
Alternatives and similar repositories for learning-to-learn-jax
Users that are interested in learning-to-learn-jax are comparing it to the libraries listed below
Sorting:
- Simple and easy stable diffusion inference with LightningModule on GPU, CPU and MPS (Possibly all devices supported by Lightning).☆17Jul 27, 2023Updated 2 years ago
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆18Nov 16, 2025Updated 3 months ago
- ☆15Dec 28, 2020Updated 5 years ago
- An easy (and fast) API for popular 3D molecular datasets!☆43Feb 27, 2026Updated last week
- Test pytorch code with minimal computational overhead☆26Jun 8, 2023Updated 2 years ago
- Introductory lectures in atomistic machine learning☆23Jul 9, 2025Updated 8 months ago
- ☆47Nov 18, 2022Updated 3 years ago
- A flexible and performant framework for training machine learning potentials.☆36Updated this week
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated 2 months ago
- ☆32Jan 7, 2024Updated 2 years ago
- ☆30Feb 11, 2022Updated 4 years ago
- Lectures on Bayesian statistics and information theory☆31May 26, 2025Updated 9 months ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆61Apr 29, 2020Updated 5 years ago
- [npj Comp. Mat.] Higher-order equivariant neural networks for charge density prediction in materials☆71Feb 21, 2025Updated last year
- Code for COVID19 CT labeling. Submillimetric CT dataset provided as well.☆13Feb 16, 2021Updated 5 years ago
- An open source implementation of CLIP.☆33Nov 7, 2022Updated 3 years ago
- Supervoxel-based Segmentation of 3D Volumetric Images☆11Feb 6, 2018Updated 8 years ago
- Neural network class for molecular dynamics to predict potential energy, forces and non-adiabatic couplings.☆11Nov 10, 2022Updated 3 years ago
- ☆10Jul 12, 2022Updated 3 years ago
- ☆35Jul 14, 2020Updated 5 years ago
- A Keras inspired training utility for PyTorch☆38Sep 13, 2018Updated 7 years ago
- Generalised UDRL☆37May 12, 2022Updated 3 years ago
- PyTorch interface for TrueGrad Optimizers☆43Aug 8, 2023Updated 2 years ago
- a new stem dataset for Music Demixing research, from the OnAir royalty-free music project☆37Mar 14, 2023Updated 2 years ago
- Repository for training and using a contrast agnostic registration model based on the work done in SynthMorph. The contrast agnostic regi…☆13Sep 5, 2023Updated 2 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Jan 16, 2026Updated last month
- Embedding language models in probability space via log-likelihood vectors☆16Oct 25, 2025Updated 4 months ago
- Web page to analyze IEEE-754 floating point values.☆19Oct 25, 2024Updated last year
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- A specification for DID create/update/deactivate operations.☆11Jan 3, 2025Updated last year
- Repository for Angus H Wright's AstroStatistics Lecture Materials☆15Apr 6, 2022Updated 3 years ago
- ☆10Jul 23, 2019Updated 6 years ago
- ☆13Dec 14, 2024Updated last year
- Implementation for ICML 2019 paper, EMI: Exploration with Mutual Information.☆37Dec 7, 2020Updated 5 years ago
- Implementation of Phase-aware speech enhancement with deep complex U-Net☆41Oct 3, 2023Updated 2 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Dec 30, 2020Updated 5 years ago
- ☆153May 25, 2020Updated 5 years ago
- Adds Algorand support to Metamask☆13Feb 25, 2024Updated 2 years ago
- Semi-Supervised Unpaired Multi-Modal Learning for Label-Efficient Medical Image Segmentation☆10Jun 29, 2021Updated 4 years ago