xitorch / xitorchLinks
Differentiable scientific computing library
☆150Updated 9 months ago
Alternatives and similar repositories for xitorch
Users that are interested in xitorch are comparing it to the libraries listed below
Sorting:
- Pytorch differentiable molecular dynamics☆176Updated 2 years ago
- Turning SymPy expressions into PyTorch modules.☆145Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆203Updated 6 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆110Updated 2 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆107Updated last month
- S2FFT: Differentiable and accelerated spherical transforms☆192Updated this week
- A parallel ODE solver for PyTorch☆255Updated 8 months ago
- Implementation of Forward Laplacian algorithm in JAX☆76Updated last week
- Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data.☆90Updated last week
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆30Updated last year
- jax library for E3 Equivariant Neural Networks☆207Updated 4 months ago
- PySCF with auto-differentiation☆79Updated 3 weeks ago
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆72Updated last year
- JAX accelerated Quantum Monte Carlo☆78Updated this week
- ☆51Updated 2 years ago
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…☆55Updated last month
- A library for programmatically generating equivariant layers through constraint solving☆268Updated 2 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆82Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆53Updated last month
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆128Updated 8 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆476Updated 2 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆53Updated 3 weeks ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆275Updated last week
- Use Jax functions in Pytorch☆242Updated last year
- Newton and Quasi-Newton optimization with PyTorch☆347Updated last week
- Exchange correlation functionals translated from libxc to jax☆45Updated 2 months ago
- Matrix-free linear algebra in JAX.☆119Updated last week
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆31Updated 11 months ago
- Boltzmann Generators and Normalizing Flows in PyTorch☆164Updated last year
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆90Updated 2 weeks ago