Fully-Differentiable Tensor Spherical Harmonics in JAX
☆21Apr 11, 2026Updated 3 months ago
Alternatives and similar repositories for tensor-spherical-harmonics
Users that are interested in tensor-spherical-harmonics are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- [ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor Products☆19Jul 16, 2025Updated last year
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆30Feb 24, 2025Updated last year
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆20Nov 16, 2025Updated 8 months ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆60Sep 26, 2025Updated 9 months ago
- An easy (and fast) API for popular 3D molecular datasets!☆51May 3, 2026Updated 2 months 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.
- ☆32Oct 9, 2025Updated 9 months ago
- Platonic Transformers: A Solid Choice For Equivariance (ICML 2026)☆62Jun 4, 2026Updated last month
- Real-time design of architectural structures with differentiable mechanics and neural networks [ICLR 2025]☆24Jun 13, 2025Updated last year
- ☆15Feb 20, 2026Updated 5 months ago
- [ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentia…☆18Jun 12, 2025Updated last year
- A symbolic regression engine for jax☆40Jul 3, 2026Updated 2 weeks ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- ☆16Jul 3, 2026Updated 2 weeks ago
- A repo for transferable sampling of molecular systems☆49Apr 27, 2026Updated 2 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- JAX library for neural operator and PDE foundation model training.☆24Updated this week
- Bridging Smoothed Molecular Dynamics and Score-Based Learning for Conformational Ensembles☆20May 16, 2026Updated 2 months ago
- Differentiable RNA folding in JAX☆30Mar 25, 2025Updated last year
- Generalised Flow Maps official implementation (ICLR 2026).☆27Oct 30, 2025Updated 8 months ago
- Geometric Generative Model Tutorial at LoG 2024☆28Nov 27, 2024Updated last year
- cinema toolkit for large data analysis and visualization☆14Sep 14, 2022Updated 3 years ago
- A PyTorch Geometric implementation of Learning to Simulate Complex Physics with Graph Networks (Sanchez-Gonzalez et al., 2020). Trains a …☆16Mar 2, 2026Updated 4 months ago
- Semi-automated grading for Marimo notebooks☆22May 8, 2026Updated 2 months ago
- Code for the paper Progressive Inference-Time Annealing of Diffusion Models for Sampling from Boltzmann Densities.☆37Jul 23, 2025Updated 11 months 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.
- Official codebase for our NeurIPS paper, Symmetry-Informed Governing Equation Discovery.☆11Nov 13, 2024Updated last year
- Framework for modified sampling from biomolecular generative models☆19Updated this week
- (ECCV 2026): Official code for Flash-BoN: Instant Drafts for Inference-Time Scaling in Diffusion Models☆17Jul 9, 2026Updated last week
- d2: Improved Techinques for Training Reasonoing Diffusion Language Models☆16Mar 25, 2026Updated 3 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆117Oct 11, 2025Updated 9 months ago
- Batched optimisation algorithms for neural network potential driven molecular dynamics.☆17Nov 27, 2025Updated 7 months ago
- ☆15Feb 5, 2025Updated last year
- Official implementation of ReDi: Rectified Discrete Flow (NeurIPS 2025)☆18May 11, 2026Updated 2 months ago
- Lightweight diffusion transformer for crystalline materials discovery☆24Updated this week
- 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.
- The Forward-Forward Algorithm for Drug Discovery☆33Dec 30, 2022Updated 3 years ago
- ☆141May 7, 2024Updated 2 years ago
- A List of all the Indian Cities with their Geodata☆16Feb 20, 2021Updated 5 years ago
- An implementation of ESM2 in Equinox+JAX☆35Apr 20, 2026Updated 3 months ago
- Vintix II: Decision Pre-Trained Transformer is a Scalable In-Context Reinforcement Learner - - = ICLR 2026☆16Apr 8, 2026Updated 3 months ago
- CUDA-accelerated radial distribution functions using NVIDIA ALCHEMI Toolkit-Ops O(N) neighbor lists and PyTorch.☆32Jun 9, 2026Updated last month
- Fast and accurate coarse-grained neural network molecular dynamics☆15Jul 13, 2026Updated last week