xitorch / xitorchLinks
Differentiable scientific computing library
☆150Updated 10 months ago
Alternatives and similar repositories for xitorch
Users that are interested in xitorch are comparing it to the libraries listed below
Sorting:
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆204Updated 7 months ago
- Turning SymPy expressions into PyTorch modules.☆148Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 3 months ago
- S2FFT: Differentiable and accelerated spherical transforms☆195Updated this week
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆108Updated 2 months ago
- Matrix-free linear algebra in JAX.☆125Updated last month
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆54Updated last month
- ☆51Updated 2 years ago
- jax library for E3 Equivariant Neural Networks☆207Updated 5 months ago
- Pytorch differentiable molecular dynamics☆176Updated 2 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆30Updated last year
- Implementation of Forward Laplacian algorithm in JAX☆76Updated 3 weeks ago
- Multi-language library for the calculation of spherical harmonics in Cartesian coordinates☆86Updated last month
- A library for programmatically generating equivariant layers through constraint solving☆268Updated 2 years ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆83Updated last year
- Orthogonal polynomials in all shapes and sizes.☆184Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆128Updated 9 months ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆96Updated last year
- A parallel ODE solver for PyTorch☆256Updated 8 months ago
- Computations involving Lie groups and harmonic analysis☆202Updated 4 months ago
- Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics☆51Updated 2 months ago
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆72Updated last year
- Automatic Functional Differentiation in JAX☆72Updated 3 months ago
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆57Updated 2 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆83Updated last year
- Newton and Quasi-Newton optimization with PyTorch☆353Updated 3 weeks 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…☆56Updated 2 months ago
- Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data.☆93Updated 3 weeks ago
- JAX accelerated Quantum Monte Carlo☆80Updated 3 weeks ago