Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)
☆29Jul 21, 2026Updated 3 months ago
Alternatives and similar repositories for SOAP_JAX
Users that are interested in SOAP_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:
- Staggered HARd-constrained Physics-Informed Neural Networks for Phase-field modelling of corrosion☆18Feb 19, 2025Updated last year
- Scale-PINN: Learning Efficient Physics-Informed Neural Networks Through Sequential Correction☆31Feb 27, 2026Updated 8 months ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19Aug 12, 2026Updated 2 months ago
- Taylor mode automatic differentiation (jets) in PyTorch☆15Updated this week
- PINNs with irreversible constraints.☆19Jun 10, 2026Updated 4 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆11Mar 14, 2024Updated 2 years ago
- ☆13Mar 21, 2024Updated 2 years ago
- Practicum on Supervised Learning in Function Spaces☆36Feb 17, 2022Updated 4 years ago
- ☆33Oct 6, 2022Updated 4 years ago
- ☆20Sep 5, 2024Updated 2 years ago
- Code for "Dual Cone Gradient Descent for Physics-Informed Neural Networks☆14Dec 6, 2025Updated 11 months ago
- This repository consists of two-step training DeepONet code.☆13May 22, 2024Updated 2 years ago
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 4 years ago
- ☆290Dec 2, 2024Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆20Aug 6, 2024Updated 2 years ago
- Morphological segmentation of Galaxies☆10Dec 17, 2021Updated 4 years ago
- ☆29Jul 17, 2025Updated last year
- Implementation of PSGD optimizer in JAX☆41Dec 31, 2024Updated last year
- Visualization of hidden layer activations of small multilayer perceptrons (MLPs)☆10Aug 26, 2024Updated 2 years ago
- CTF workshop deep learning tutorial and hackathon☆13Feb 16, 2023Updated 3 years ago
- TANTE: Time-Adaptive Operator Learning via Neural Taylor Expansion☆18Jul 3, 2026Updated 4 months ago
- ☆13Jun 10, 2025Updated last year
- ☆16Oct 25, 2021Updated 5 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.
- LaTeX style file for Transactions on Machine Learning Research☆17Jun 30, 2023Updated 3 years ago
- ☆11Jan 15, 2026Updated 9 months ago
- [Poster; ICLR 2026] [Oral; Neurips OPT2024] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Apr 15, 2026Updated 6 months ago
- Repository for the paper "Bayesian Quadrature on Riemannian Data Manifolds"☆10Jun 18, 2024Updated 2 years ago
- Near-linear scaling neural PDE solver for large unstructured meshes via multi-scale attention with ball-tree partitioning☆15May 10, 2026Updated 5 months ago
- A package for computing matrix exponentials and finite horizon Gramians☆11Jan 21, 2026Updated 9 months ago
- HANDY - Handy tool for spectra normalization☆11Nov 12, 2020Updated 5 years ago
- ☆70Jul 10, 2023Updated 3 years ago
- Convolutions and more as einsum for PyTorch☆18Jun 6, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Physics-informed neural networks in JAX — multi-GPU training, advanced PINN algorithms, and 16 benchmark examples☆35Aug 12, 2026Updated 2 months ago
- ☆12Dec 15, 2022Updated 3 years ago
- Calculates the eigendecomposition of the Laplace-Beltrami operator on charts of Riemannian manifolds. Moved to https://git.superadditive.…☆10Dec 31, 2019Updated 6 years ago
- Physics-informed neural networks for highly compressible flows 🧠🌊☆30Nov 26, 2023Updated 2 years ago
- ISMIR 2021: Curriculum Learning for Imbalanced Classification in Large Vocabulary Automatic Chord Recognition☆10Nov 8, 2021Updated 4 years ago
- ACCO: An optimization algorithm for sharded distributed LLM training.☆13May 22, 2025Updated last year
- ☆76Oct 5, 2026Updated last month