Vis4SciML / LandscaperLinks
Landscaper is a comprehensive Python framework designed for exploring the loss landscapes of deep learning models.
☆20Updated last month
Alternatives and similar repositories for Landscaper
Users that are interested in Landscaper are comparing it to the libraries listed below
Sorting:
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- ☆53Updated 3 years ago
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆44Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated 2 years ago
- A collection of graph neural networks implementations in JAX☆35Updated 2 years ago
- TorchFSM: Fourier Spectral Method with PyTorch☆53Updated last week
- Exact OU processes with JAX☆59Updated this week
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆90Updated last year
- ☆17Updated last year
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Implementation of various equivariant models in JAX☆12Updated last year
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆46Updated 2 years ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆109Updated 3 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆32Updated 2 years ago
- The Forward-Forward Algorithm for Drug Discovery☆34Updated 3 years ago
- ☆169Updated last year
- ⚡ Triton implementation of Clifford algebra neural networks.☆32Updated 2 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 5 months ago
- ☆41Updated 2 years ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆51Updated 8 months ago
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆81Updated 5 years ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆25Updated last year
- Code for paper E(n)-Equivariant Topological Neural Networks☆36Updated last year
- Stencil computations in JAX☆71Updated last month
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆45Updated 2 years ago
- Steerable E(3) GNN in jax☆24Updated 2 years ago
- Graph Convolutional Networks in JAX