E(n) Equivariant GNN in jax
☆14Aug 31, 2023Updated 3 years ago
Alternatives and similar repositories for egnn-jax
Users that are interested in egnn-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:
- PaiNN in jax☆11Jan 14, 2025Updated last year
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 3 years ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆44Mar 1, 2025Updated last year
- a differentiable lensing simulator☆12Jan 18, 2022Updated 4 years ago
- ☆16May 12, 2026Updated 5 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Investigating the information content in the cosmic web through dark matter halo graphs. Arxiv paper: https://arxiv.org/abs/2207.05202☆14Jan 18, 2024Updated 2 years ago
- [ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor Products☆20Jul 16, 2025Updated last year
- Cosmological map inference with deep learning☆18Nov 3, 2021Updated 5 years ago
- Graph Networks with Jax☆11Mar 23, 2022Updated 4 years ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- Generative materials benchmarking metrics, inspired by guacamol and CDVAE.☆41Jun 1, 2024Updated 2 years ago
- ☆13Dec 13, 2023Updated 2 years ago
- Accelerated Bayesian SED modeling using Amortized Neural Posterior Estimation☆25Dec 11, 2025Updated 10 months ago
- ☆15Dec 28, 2020Updated 5 years 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.
- Collection of TensorFlow modules for GalSim☆10Jul 27, 2021Updated 5 years ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆21Jan 18, 2025Updated last year
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 3 years ago
- A python package for making arbitrarily structured, arbitrary-dimension boxes☆29Updated this week
- variational free-energy of dense hydrogen☆14Sep 25, 2023Updated 3 years ago
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 4 years ago
- Differentiable models of halo and galaxy formation history☆11Jul 1, 2026Updated 4 months ago
- A Jax package to perform Neural Density Estimation from cosmological probes.☆13Jun 19, 2026Updated 4 months ago
- CrystalGRW: Generative Modeling of Crystal Structures with Targeted Properties via Geodesic Random Walks☆15May 22, 2026Updated 5 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- RealSim is the statistical observational realism suite described in Bottrell et al 2017ab and made public in Bottrell et al 2019b.☆12Jul 7, 2021Updated 5 years ago
- caskade manages simulator parameters for you☆14Jul 27, 2026Updated 3 months ago
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- Code for the paper 'Neural Variational Gradient Descent'. We perform nonparametric variational inference by transporting samples along a …☆12Jul 29, 2021Updated 5 years ago
- Repository hosting code for creating deep generative models for GalSim☆13Aug 11, 2020Updated 6 years ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- ☆16May 30, 2023Updated 3 years ago
- A JAX package for weak gravitational lensing☆36Apr 30, 2024Updated 2 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.1482…☆12Feb 26, 2023Updated 3 years ago
- Free-form gravitational lensing reconstruction of density maps and background intensity maps using a Recurrent Inference Machine.☆11Jan 26, 2023Updated 3 years ago
- Repository for course offered by the Stack Club☆14Jan 13, 2023Updated 3 years ago
- ☆35Sep 12, 2022Updated 4 years ago
- Photometric error model for astronomical imaging surveys☆12Sep 9, 2026Updated last month
- ☆17Jul 18, 2023Updated 3 years ago
- TJPCov is a general covariance calculator interface to be used within LSST DESC☆15Updated this week