A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).
☆26Nov 18, 2025Updated 11 months ago
Alternatives and similar repositories for enf-min-jax
Users that are interested in enf-min-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:
- ☆16Jan 23, 2025Updated last year
- ☆12Jan 9, 2024Updated 2 years ago
- Code for reproducing the paper "Space-Time Continuous PDE Forecasting using Equivariant Neural Fields" (https://arxiv.org/abs/2406.06660)…☆16Nov 18, 2025Updated 11 months ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆44Mar 1, 2025Updated last year
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆94May 30, 2024Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated last year
- ☆14Aug 8, 2023Updated 3 years ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆20Jan 18, 2025Updated last year
- ☆30May 31, 2026Updated 4 months ago
- Official JAX implementation of neural isometries - taming transformations for equivariant ML☆36Aug 1, 2025Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- Code repository of the paper "Exploiting Redundancy: Separable Group Convolutional Networks on Lie Groups" https://proceedings.mlr.press/…☆11Jul 20, 2022Updated 4 years ago
- A pytorch toolbox designed for experimentation using a generalization of the Resnet design.☆15Jul 6, 2023Updated 3 years ago
- Generative models with JAX.☆22Feb 21, 2026Updated 7 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆120Oct 11, 2025Updated last year
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆26Jan 3, 2024Updated 2 years ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆142Dec 20, 2023Updated 2 years ago
- MARIO: Multiscale Aerodynamic Resolution Invariant Operator. 3rd Place at the ML4CFD Challenge at Neurips 2024.☆22Oct 8, 2025Updated last year
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆31Jan 22, 2024Updated 2 years ago
- [NeurIPS 2022] "Signal Processing for Implicit Neural Representations" by Dejia Xu*, Peihao Wang*, Yifan Jiang, Zhiwen Fan, Zhangyang Wan…☆53Dec 1, 2022Updated 3 years ago
- Differentiable Likelihood for CMB Analysis☆32Jun 5, 2026Updated 4 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆33Jun 28, 2023Updated 3 years ago
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- JAX-based linear Einstein-Boltzmann solver for cosmology☆29Nov 13, 2025Updated 11 months ago
- Hierarchy of parameterized Smoothed Particle Hydrodynamics models trained with mixed mode AD and Sensitivity Analysis (SA)☆13Sep 9, 2022Updated 4 years ago
- Benchmarking and validation of Machine Learning Interatomic Potentials (MLIP)☆19Sep 30, 2026Updated 2 weeks ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆30Feb 24, 2025Updated last year
- Code for ICML 24 paper "Implicit Representations via Operator Learning"☆13Jul 30, 2024Updated 2 years ago
- [GSI 2023] Learning Lagrangian Fluid Mechanics with E(3)-Equivariant GNNs☆15Jun 3, 2024Updated 2 years ago
- Implementation of the algorithms in the research paper iNeRF☆10Oct 1, 2021Updated 5 years ago
- Neural SPH☆48Nov 23, 2025Updated 10 months ago
- Python API for the EAGLE cosmological simulation database.☆12Mar 15, 2023Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- PLAID (Physics-Learning AI Datamodel), a flexible and extensible framework for representing and sharing datasets of physics simulations☆31Oct 1, 2026Updated 2 weeks ago
- Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)☆51Updated this week
- Benchmarking field-level cosmological inference from galaxy surveys.☆16Jul 17, 2025Updated last year
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 3 years ago
- ☆29Jul 17, 2025Updated last year
- [ICML 2026] Official code release for paper "Temporal Score Rescaling for Temperature Sampling in Diffusion and Flow Models"☆54Jun 12, 2026Updated 4 months ago
- ☆10Dec 17, 2019Updated 6 years ago