Alternatives and similar repositories for nvcc-feedstock:

Users that are interested in nvcc-feedstock are comparing it to the libraries listed below

• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆63Updated 6 months ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆41Updated 4 years ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 4 years ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆63Updated 3 years ago
• SymJAX / SymJAX
  Documentation:
  ☆119Updated last year
• QuantStack / xsci
  Scientific algorithms implemented on top of the x-stack (xtensor, xsimd ...)
  ☆9Updated 5 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• dfm / jpu
  JAX + Units
  ☆44Updated 3 weeks ago
• google-deepmind / einshape
  ☆100Updated 6 months ago
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• SamDuffield / mocat
  All things Monte Carlo, written in JAX.
  ☆30Updated last year
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆13Updated 7 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆43Updated last week
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆43Updated last week
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆25Updated 4 months ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆44Updated last week
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• geomstats / applications
  Examples of more involved applications using Geomstats
  ☆31Updated 3 years ago
• MilesCranmer / python_citations
  Bibtex for various Python science and machine learning software
  ☆32Updated 2 years ago
• mattjj / ode-diff-notes
  ☆12Updated 4 years ago
• tmbdev-archive / tarproc
  Utilities for sequential processing of tar files.
  ☆24Updated 2 years ago
• cgarciae / einop
  ☆58Updated 2 years ago
• Quansight-Labs / udiff
  Automatic differentiation with uarray/unumpy.
  ☆16Updated 3 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• duvenaud / jaxde
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Updated 4 years ago
• NIFTy-PPL / NIFTy
  Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions
  ☆22Updated last week