Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024
☆128Nov 27, 2024Updated last year
Alternatives and similar repositories for stde
Users that are interested in stde are comparing it to the libraries listed below
Sorting:
- ☆13Jul 25, 2023Updated 2 years ago
- Exchange correlation functionals translated from libxc to jax☆52Mar 24, 2025Updated 11 months ago
- Automatic Functional Differentiation in JAX☆81Sep 18, 2025Updated 5 months ago
- an environment based on XLA for deep learning compiler optimization research.☆24Mar 7, 2023Updated 3 years ago
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated 2 months ago
- Intriguing Properties of Data Attribution on Diffusion Models (ICLR 2024)☆37Jan 23, 2024Updated 2 years ago
- ☆32Feb 14, 2024Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- Repository for some of the experiments presented in the paper "Deep Learning Alternatives of the Kolmogorov Superposition Theorem", accep…☆22Mar 28, 2025Updated 11 months ago
- A JAX library for Density Functional Theory.☆54Nov 25, 2025Updated 3 months ago
- Spectral Neural Operator☆79Dec 20, 2023Updated 2 years ago
- This repository contains the code and models for our paper "Investigating and Mitigating Failure Modes in Physics-informed Neural Network…☆20Dec 8, 2023Updated 2 years ago
- Source code for paper Conservative Uncertainty Estimation By Fitting Prior Networks (ICLR 2020)☆22Nov 28, 2022Updated 3 years ago
- ☆20Sep 1, 2025Updated 6 months ago
- Code for DCPS algorithm, NeurIPS 2024, https://arxiv.org/abs/2403.11407☆11Dec 11, 2024Updated last year
- The official implementation of paper "Drop-Activation: Implicit Parameter Reduction and Harmonious Regularization".☆10May 30, 2019Updated 6 years ago
- Hearst Patterns to extract Hypernyms from text☆13Oct 30, 2019Updated 6 years ago
- Tackling the Curse of Dimensionality with Physics-Informed Neural Networks☆15May 17, 2024Updated last year
- Code and data for paper named: Large language models for automatic equation discovery of nonlinear dynamics☆12Mar 6, 2025Updated last year
- Jax implementation of VIT-VQGAN☆10Jan 25, 2024Updated 2 years ago
- ☆17Apr 14, 2025Updated 10 months ago
- ☆11Jan 28, 2023Updated 3 years ago
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆75Mar 15, 2024Updated last year
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆48Jan 31, 2024Updated 2 years ago
- Official implementation of Bootstrapping Language Models via DPO Implicit Rewards☆47Apr 15, 2025Updated 10 months ago
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆29May 30, 2025Updated 9 months ago
- ☆55Sep 17, 2024Updated last year
- Official code implementation for the ACL 2024 Student Research Workshop paper "In-Context Symbolic Regression: Leveraging Large Language …☆17Sep 26, 2024Updated last year
- ☆13Nov 22, 2024Updated last year
- Implementation of the work Variational multiple shooting for Bayesian ODEs with Gaussian processes☆13Aug 5, 2022Updated 3 years ago
- ☆13Jul 12, 2024Updated last year
- Implementation for "DeltaPhi: Learning Physical Trajectory Residual for PDE Solving"☆13Jun 17, 2024Updated last year
- [TMLR] CodePDE: An Inference Framework for LLM-driven PDE Solver Generation☆71Feb 12, 2026Updated 3 weeks ago
- A python implementation of spherical Bessel transform (SBT) based on algorithm proposed by J. Talman.☆13Dec 18, 2023Updated 2 years ago
- ☆56May 13, 2025Updated 9 months ago
- Official implementation of Scalable Transformer for PDE surrogate modelling☆54Mar 23, 2024Updated last year
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆521Updated this week
- Muon fsdp 2☆55Aug 8, 2025Updated 6 months ago
- Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.☆27Aug 9, 2024Updated last year