Alternatives and similar repositories for xitorch:

Users that are interested in xitorch are comparing it to the libraries listed below

• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆107Updated last month
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆116Updated 3 months ago
• patrick-kidger / sympytorch
  Turning SymPy expressions into PyTorch modules.
  ☆149Updated last year
• e3nn / e3nn-jax
  jax library for E3 Equivariant Neural Networks
  ☆204Updated 2 months ago
• esa / torchquad
  Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
  ☆204Updated 4 months ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆125Updated 7 months ago
• torchmd / mdgrad
  Pytorch differentiable molecular dynamics
  ☆175Updated 2 years ago
• google-research / e3x
  E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
  ☆103Updated last week
• google-deepmind / flows_for_atomic_solids
  ☆49Updated 2 years ago
• sigma-py / orthopy
  Orthogonal polynomials in all shapes and sizes.
  ☆184Updated last year
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆266Updated last year
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆476Updated last month
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆94Updated last year
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆71Updated 3 weeks ago
• astro-informatics / s2fft
  Differentiable and accelerated spherical transforms with JAX
  ☆147Updated this week
• martenlienen / torchode
  A parallel ODE solver for PyTorch
  ☆254Updated 6 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆51Updated last week
• zlin7 / CGNet
  ☆69Updated 2 years ago
• YWolfeee / lapjax
  A JAX based package designed for efficient second order operators (e.g., laplacian) computation.
  ☆72Updated last year
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆267Updated this week
• microsoft / folx
  Implementation of Forward Laplacian algorithm in JAX
  ☆63Updated last month
• vdutor / SphericalHarmonics
  Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
  ☆28Updated 11 months ago
• gngdb / pytorch-minimize
  Use scipy.optimize.minimize as a PyTorch Optimizer.
  ☆72Updated 8 months ago
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆333Updated 2 months ago
• lab-cosmo / sphericart
  Multi-language library for the calculation of spherical harmonics in Cartesian coordinates
  ☆83Updated last week
• rfeinman / pytorch-minimize
  Newton and Quasi-Newton optimization with PyTorch
  ☆342Updated last year
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆76Updated 3 years ago
• patrick-kidger / torchcubicspline
  Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…
  ☆249Updated 2 years ago
• patrick-kidger / lineax
  Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
  ☆431Updated 3 weeks ago
• bytedance / jaqmc
  JAX accelerated Quantum Monte Carlo
  ☆74Updated last month