Fully-Differentiable Tensor Spherical Harmonics in JAX
☆21Apr 11, 2026Updated 5 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☆20Jul 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☆22Nov 16, 2025Updated 10 months ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆60Sep 26, 2025Updated last year
- An easy (and fast) API for popular 3D molecular datasets!☆50May 3, 2026Updated 5 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆37Oct 9, 2025Updated last year
- Real-time design of architectural structures with differentiable mechanics and neural networks [ICLR 2025]☆24Jun 13, 2025Updated last year
- Platonic Transformers: A Solid Choice For Equivariance (ICML 2026)☆66Sep 25, 2026Updated 2 weeks ago
- The AI Streamliner Project☆24Sep 11, 2025Updated last year
- [ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget☆79Apr 5, 2026Updated 6 months ago
- ☆17Updated this week
- A symbolic regression engine for jax☆43Sep 26, 2026Updated 2 weeks ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- CASP15 performance benchmarking of the state-of-the-art protein structure prediction methods☆16Dec 13, 2023Updated 2 years 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.
- ☆17Aug 7, 2026Updated 2 months ago
- Bridging Smoothed Molecular Dynamics and Score-Based Learning for Conformational Ensembles☆22May 16, 2026Updated 4 months ago
- Generalised Flow Maps official implementation (ICLR 2026).☆28Oct 30, 2025Updated 11 months ago
- cinema toolkit for large data analysis and visualization☆14Sep 14, 2022Updated 4 years ago
- Semi-automated grading for Marimo notebooks☆21Updated this week
- Code for the paper Progressive Inference-Time Annealing of Diffusion Models for Sampling from Boltzmann Densities.☆38Sep 5, 2026Updated last month
- Framework for modified sampling from biomolecular generative models☆28Updated this week
- UBio-MolFM is a foundation model suite for molecular modeling, developed by the UBio-MolFM team.☆48Aug 13, 2026Updated last month
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆120Oct 11, 2025Updated 11 months ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Batched optimisation algorithms for neural network potential driven molecular dynamics.☆17Nov 27, 2025Updated 10 months ago
- ☆15Feb 5, 2025Updated last year
- ☆143May 7, 2024Updated 2 years ago
- The Forward-Forward Algorithm for Drug Discovery☆33Dec 30, 2022Updated 3 years ago
- Official implementation of ReDi: Rectified Discrete Flow (NeurIPS 2025)☆19May 11, 2026Updated 4 months ago
- (ECCV 2026): Official code for Flash-BoN: Instant Drafts for Inference-Time Scaling in Diffusion Models☆25Jul 9, 2026Updated 3 months ago
- Vintix II: Decision Pre-Trained Transformer is a Scalable In-Context Reinforcement Learner - - = ICLR 2026☆17Apr 8, 2026Updated 6 months ago
- CUDA-accelerated radial distribution functions using NVIDIA ALCHEMI Toolkit-Ops O(N) neighbor lists and PyTorch.☆34Jun 9, 2026Updated 4 months ago
- ☆36May 15, 2026Updated 4 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.
- Fast and accurate coarse-grained neural network molecular dynamics☆16Jul 13, 2026Updated 2 months ago
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 3 years ago
- Differentiable Flexible Mechanical Metamaterials☆41Sep 22, 2026Updated 2 weeks ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆61Mar 26, 2025Updated last year
- Next generation machine learning force field on enzymatic catalysis☆18Updated this week
- Jax-based non-linear reconciliation and learning☆17Jun 25, 2026Updated 3 months ago
- Zero Shot Molecular Generation via Similarity Kernels☆31Aug 30, 2026Updated last month