xitorch / xitorch
Differentiable scientific computing library
☆136Updated last month
Related projects: ⓘ
- Turning SymPy expressions into PyTorch modules.☆139Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆78Updated 2 weeks ago
- ☆48Updated last year
- jax library for E3 Equivariant Neural Networks☆175Updated last month
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆88Updated 2 weeks ago
- Differentiable and accelerated spherical transforms with JAX☆123Updated this week
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆182Updated 2 months ago
- Pytorch differentiable molecular dynamics☆168Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆119Updated this week
- A parallel ODE solver for PyTorch☆218Updated 3 weeks ago
- A library for programmatically generating equivariant layers through constraint solving☆252Updated last year
- PySCF with auto-differentiation☆63Updated last week
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆63Updated last month
- Abstract your array operations.☆132Updated 4 months ago
- This is project page for the paper "RG-Flow: a hierarchical and explainable flow model based on renormalization group and sparse prior". …☆83Updated 2 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆87Updated 10 months ago
- Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data.☆81Updated last month
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆68Updated 3 months ago
- Differentiable Quantum Chemistry (only Differentiable Density Functional Theory and Hartree Fock at the moment)☆106Updated 2 years ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆421Updated this week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆231Updated this week
- Turn SymPy expressions into trainable JAX expressions.☆313Updated 5 months ago
- Orthogonal polynomials in all shapes and sizes.☆182Updated 7 months ago
- Multi-language library for the calculation of spherical harmonics in Cartesian coordinates☆69Updated this week
- Newton and Quasi-Newton optimization with PyTorch☆306Updated 6 months ago
- Equivariant machine learning interatomic potentials in JAX.☆59Updated 11 months ago
- Exchange correlation functionals translated from libxc to jax☆42Updated 9 months ago
- Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics☆45Updated 9 months ago
- Normalizing Flows using JAX☆82Updated 9 months ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆340Updated 2 weeks ago