JAX implementation of Learning to learn by gradient descent by gradient descent
☆28Aug 5, 2025Updated 8 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Pure C implementation of e3nn☆24Mar 17, 2025Updated last year
- An ASE-friendly implementation of the amorphous-to-crystalline (a2c) workflow.☆18Oct 19, 2025Updated 6 months ago
- 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
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated last month
- This repository gathers the SchNet4AIM code along with some instructions and readme files.☆15Mar 13, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Calculate observables from neural network-based VMC (NN-VMC).☆17Apr 16, 2025Updated last year
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆19Nov 16, 2025Updated 5 months ago
- ☆22May 7, 2025Updated 11 months ago
- An easy (and fast) API for popular 3D molecular datasets!☆48Mar 24, 2026Updated 3 weeks ago
- ☆15Dec 28, 2020Updated 5 years ago
- Introductory lectures in atomistic machine learning☆24Jul 9, 2025Updated 9 months ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆79Jan 7, 2026Updated 3 months ago
- A flexible and performant framework for training machine learning potentials.☆36Updated this week
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Sep 29, 2021Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget and [arXiv'26] Phonon fine-tuning (PFT)☆73Apr 5, 2026Updated 2 weeks ago
- [npj Comp. Mat.] Higher-order equivariant neural networks for charge density prediction in materials☆72Feb 21, 2025Updated last year
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆15May 28, 2025Updated 10 months ago
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Feb 20, 2023Updated 3 years ago
- An open source implementation of CLIP.☆33Nov 7, 2022Updated 3 years ago
- gzip middleware for ASGI applications, extracted from Starlette☆12Apr 9, 2026Updated last week
- PyTorch interface for TrueGrad Optimizers☆43Aug 8, 2023Updated 2 years ago
- Rosetta Server for Solana Blockchain☆11Feb 17, 2021Updated 5 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆61Apr 29, 2020Updated 5 years ago
- Serverless GPU API endpoints on Runpod - Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆44May 3, 2024Updated last year
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated last year
- Blog of the LibreCV.org☆11May 17, 2021Updated 4 years ago
- 7th place solution☆22Mar 28, 2021Updated 5 years ago
- a differentiable lensing simulator☆10Jan 18, 2022Updated 4 years ago
- Agar.io for Continual Reinforcement Learning☆24Jul 24, 2025Updated 8 months ago
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated last year
- Python re-implementation of szabo.f☆10Jul 30, 2015Updated 10 years ago
- PaiNN in jax☆11Jan 14, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Implementation of LogAvgExp for Pytorch☆37Apr 10, 2025Updated last year
- ☆26Jul 19, 2019Updated 6 years ago
- Nomalizing flows for orbita-free DFT☆11Sep 20, 2024Updated last year
- ☆32Jan 7, 2024Updated 2 years ago
- New TEAL ops in AVM6☆11Apr 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
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago