Vis4SciML / LandscaperLinks
Landscaper is a comprehensive Python framework designed for exploring the loss landscapes of deep learning models.
☆20Updated 2 weeks ago
Alternatives and similar repositories for Landscaper
Users that are interested in Landscaper are comparing it to the libraries listed below
Sorting:
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated 2 years ago
- ☆53Updated 3 years ago
- A collection of graph neural networks implementations in JAX☆34Updated 2 years ago
- Implementation of various equivariant models in JAX☆12Updated last year
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- Exact OU processes with JAX☆58Updated 9 months ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆89Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 4 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆32Updated 2 years ago
- Implementation of approximate free-energy minimization in PyTorch☆20Updated 4 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- ☆167Updated last year
- Steerable E(3) GNN in jax☆24Updated 2 years ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆45Updated 2 years ago
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆81Updated 5 years ago
- Stencil computations in JAX☆71Updated last month
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆54Updated 8 months ago
- Implementation of Forward Laplacian algorithm in JAX☆92Updated 3 weeks ago
- ☆41Updated 2 years ago
- Pytorch implement of the paper Neural Canonical Transformation with Symplectic Flows☆31Updated 5 years ago
- ☆17Updated last year
- Graph Convolutional Networks in JAX☆33Updated 4 years ago
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆44Updated last year
- ☆31Updated last year
- Multiple dispatch over abstract array types in JAX.☆137Updated this week
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year