haydn-jones / SOAP_JAXView external linksLinks
Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)
☆24Jan 9, 2026Updated last month
Alternatives and similar repositories for SOAP_JAX
Users that are interested in SOAP_JAX are comparing it to the libraries listed below
Sorting:
- This repository consists of two-step training DeepONet code.☆14May 22, 2024Updated last year
- ☆16Aug 6, 2024Updated last year
- Transformed Generative Pre-Trained Physics-Informed Neural Networks (TGPT-PINN), a framework that extends Physics-Informed Neural Network…☆18Apr 10, 2024Updated last year
- ☆12Aug 22, 2025Updated 5 months ago
- ☆53Jul 21, 2025Updated 6 months ago
- ☆59Jul 10, 2023Updated 2 years ago
- Benchmarking of diffusion models for global field reconstruction from sparse observations☆30Dec 4, 2024Updated last year
- ☆30Oct 6, 2022Updated 3 years ago
- simple bibtex generator for any text with \cite{}☆31Jul 13, 2024Updated last year
- ☆16Nov 2, 2025Updated 3 months ago
- This project is divided in a two parts. In first study, Lame parameters are identified using tanh activation function. After that, six a…☆13Nov 17, 2022Updated 3 years ago
- ☆250Dec 2, 2024Updated last year
- ☆61Feb 3, 2026Updated last week
- Code for "Accurate Differential Operators for Hybrid Neural Fields", accepted at CVPR 2025☆28Jun 5, 2025Updated 8 months ago
- Implementation of PSGD optimizer in JAX☆35Dec 31, 2024Updated last year
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Mar 21, 2023Updated 2 years ago
- ☆17Apr 18, 2024Updated last year
- [CVPR 2020] A generative model with latent factors that are independent and localized.☆12Mar 27, 2025Updated 10 months ago
- Call XGBoost from Fortran☆12Jun 20, 2023Updated 2 years ago
- Numerical assessments of a nonintrusive surrogate model based on recurrent neural networks and proper orthogonal decomposition: Rayleigh …☆10Dec 2, 2022Updated 3 years ago
- Practicum on Supervised Learning in Function Spaces☆35Feb 17, 2022Updated 3 years ago
- [AAAI 2023] Physics-Informed Cell Representations☆36Nov 27, 2025Updated 2 months ago
- ☆35Nov 22, 2024Updated last year
- ☆11Mar 31, 2021Updated 4 years ago
- 深度学习之神经网络核心原理与算法-课程学习相关代码☆12Aug 18, 2018Updated 7 years ago
- Official Code for ICML 2024 paper "TENG: Time-Evolving Natural Gradient for Solving PDEs With Deep Neural Nets Toward Machine Precision"☆18Nov 18, 2024Updated last year
- KTH-FlowAI / Towards-extraction-of-orthogonal-and-parsimonious-non-linear-modes-from-turbulent-flows☆11Apr 5, 2023Updated 2 years ago
- Brief tutorial how to install linux on Asus T100HA☆13May 2, 2017Updated 8 years ago
- Official code for AL-PINNS: Augmented Lagrangian relaxation method for Physics-Informed Neural Networks☆12Jul 29, 2023Updated 2 years ago
- ☆10Mar 8, 2023Updated 2 years ago
- Source code for the paper "Sample-efficient inverse design of freeform nanophotonic devices with physics-informed reinforcement learning"☆12Apr 12, 2024Updated last year
- Physics Informed Sparse Identification of Nonlinear Dynamics☆12Jan 3, 2025Updated last year
- Code for paper "Beyond Closure Models: Learning Chaotic Systems via Physics-Informed Neural Operators".☆14Dec 24, 2025Updated last month
- ☆49Aug 19, 2025Updated 5 months ago
- ACCO: An optimization algorithm for sharded distributed LLM training.☆13May 22, 2025Updated 8 months ago
- Neural Operators with Applications to the Helmholtz Equation☆11Sep 19, 2024Updated last year
- A fine-mapping method integrating GWAS summary statistics and functional annotation data☆11Dec 28, 2023Updated 2 years ago
- This repository is a guided tutorial on Pseudo-Spectral method that in long run can help as a basis for study material on Direct Numerica…☆12Sep 23, 2021Updated 4 years ago