conda-forge / nvcc-feedstock

Related projects: ⓘ

• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆60Updated 2 months ago
• google-deepmind / einshape
  ☆97Updated 2 months ago
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated last year
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆36Updated last week
• francois-rozet / torchist
  NumPy-style histograms in PyTorch
  ☆51Updated 5 months ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆44Updated this week
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆59Updated last year
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆29Updated 4 years ago
• JoeriHermans / awflow
  Reproducible research and reusable acyclic workflows in Python. Execute code on HPC systems as if you executed them on your personal comp…
  ☆18Updated 2 years ago
• tianluyuan / sphere
  Implementation of FB8, a generalization of the Kent (1982) and Bingham-Mardia (1978) distributions on a sphere
  ☆16Updated last week
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 2 years ago
• b45ch1 / algopy
  AlgoPy is a Research Prototype for Algorithmic Differentation in Python
  ☆79Updated 2 months ago
• SymJAX / SymJAX
  Documentation:
  ☆119Updated last year
• vicariousinc / PGMax
  Loopy belief propagation for factor graphs on discrete variables, in JAX!
  ☆62Updated 2 months ago
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆34Updated last month
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆39Updated 11 months ago
• jax-ml / jax-ai-stack
  ☆28Updated this week
• Linux-cpp-lisp / opt_einsum_fx
  Einsum optimization using opt_einsum and PyTorch FX graph rewriting
  ☆19Updated 2 years ago
• glouppe / flowing-with-jax
  ☆15Updated 3 years ago
• mattjj / ode-diff-notes
  ☆12Updated 3 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆76Updated last year
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆40Updated 4 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆41Updated 3 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 3 years ago
• NVIDIA / numba-cuda
  ☆28Updated this week
• scipy / oldest-supported-numpy
  Meta-package providing the oldest supported Numpy for a given Python version and platform
  ☆54Updated this week
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆47Updated this week