xitorch / xitorch
Differentiable scientific computing library
☆148Updated 7 months ago
Alternatives and similar repositories for xitorch:
Users that are interested in xitorch are comparing it to the libraries listed below
- Turning SymPy expressions into PyTorch modules.☆146Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆106Updated 2 weeks ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated 4 months ago
- A library for programmatically generating equivariant layers through constraint solving☆266Updated last year
- ☆49Updated 2 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆28Updated 10 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆200Updated 4 months ago
- Matrix-free linear algebra in JAX.☆116Updated 2 months 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…☆54Updated 3 months ago
- jax library for E3 Equivariant Neural Networks☆202Updated 2 months ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆81Updated 9 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆123Updated 6 months ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆93Updated last year
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- Pytorch differentiable molecular dynamics☆175Updated 2 years ago
- Differentiable and accelerated spherical transforms with JAX☆146Updated this week
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆199Updated this week
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆80Updated 9 months ago
- A parallel ODE solver for PyTorch☆251Updated 5 months ago
- Implementation of Forward Laplacian algorithm in JAX☆61Updated last month
- ☆69Updated 2 years ago
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆71Updated last year
- Exchange correlation functionals translated from libxc to jax☆45Updated this week
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆83Updated this week
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆70Updated 8 months ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆70Updated 10 months ago
- Automatic Functional Differentiation in JAX☆67Updated this week
- Mathematical operations for JAX pytrees☆198Updated 3 months ago
- Boltzmann Generators and Normalizing Flows in PyTorch☆160Updated last year
- A software package for flexible HPC GPs☆16Updated last month