Alternatives and similar repositories for tnpy

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

• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,583Updated last month
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,436Updated 2 months ago
• dfm / extending-jax
  Extending JAX with custom C++ and CUDA code
  ☆400Updated last year
• patrick-kidger / torchtyping
  Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
  ☆1,454Updated 6 months ago
• google / fiddle
  ☆359Updated this week
• merantix-momentum / daain
  This is the code for our paper DAAIN: Detection of Anomalous and AdversarialInput using Normalizing Flows
  ☆15Updated 4 years ago
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,089Updated 3 weeks ago
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆975Updated last week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,064Updated last year
• dgasmith / opt_einsum
  ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
  ☆953Updated 4 months ago
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,138Updated 2 weeks ago
• meta-pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,230Updated last week
• facebookincubator / submitit
  Python 3.8+ toolbox for submitting jobs to Slurm
  ☆1,531Updated 3 weeks ago
• scikit-build / scikit-build-core
  A next generation Python CMake adaptor and Python API for plugins
  ☆404Updated last week
• nv-legate / cupynumeric
  NumPy and SciPy on Multi-Node Multi-GPU systems
  ☆935Updated last week
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆698Updated 2 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,073Updated last month
• wjakob / nanobind
  nanobind: tiny and efficient C++/Python bindings
  ☆3,102Updated this week
• brentyi / tyro
  CLI interfaces & config objects, from types
  ☆814Updated this week
• mit-ll-responsible-ai / hydra-zen
  Create powerful Hydra applications without the yaml files and boilerplate code.
  ☆419Updated this week
• BlackHC / toma
  Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
  ☆437Updated last year
• EnzymeAD / Enzyme
  High-performance automatic differentiation of LLVM and MLIR.
  ☆1,481Updated this week
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆330Updated 3 years ago
• google-deepmind / chex
  ☆888Updated 2 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆2,065Updated this week
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆991Updated this week
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆883Updated this week
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆1,001Updated 3 weeks ago
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆439Updated this week
• coreylowman / dfdx
  Deep learning in Rust, with shape checked tensors and neural networks
  ☆1,855Updated last year