tdakhran / tnpyLinks
C++17 numpy npy library
☆17Updated 4 months ago
Alternatives and similar repositories for tnpy
Users that are interested in tnpy are comparing it to the libraries listed below
Sorting:
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,583Updated last month
 - functorch is JAX-like composable function transforms for PyTorch.☆1,436Updated 2 months ago
 - Extending JAX with custom C++ and CUDA code☆400Updated last year
 - Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,454Updated 6 months ago
 - ☆359Updated this week
 - This is the code for our paper DAAIN: Detection of Anomalous and AdversarialInput using Normalizing Flows☆15Updated 4 years ago
 - common in-memory tensor structure☆1,089Updated 3 weeks ago
 - TensorDict is a pytorch dedicated tensor container.☆975Updated last week
 - A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,064Updated last year
 - ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆953Updated 4 months ago
 - KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,138Updated 2 weeks ago
 - A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,230Updated last week
 - Python 3.8+ toolbox for submitting jobs to Slurm☆1,531Updated 3 weeks ago
 - A next generation Python CMake adaptor and Python API for plugins☆404Updated last week
 - NumPy and SciPy on Multi-Node Multi-GPU systems☆935Updated last week
 - PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆698Updated 2 years ago
 - Profiling and inspecting memory in pytorch☆1,073Updated last month
 - nanobind: tiny and efficient C++/Python bindings☆3,102Updated this week
 - CLI interfaces & config objects, from types☆814Updated this week
 - Create powerful Hydra applications without the yaml files and boilerplate code.☆419Updated this week
 - Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆437Updated last year
 - High-performance automatic differentiation of LLVM and MLIR.☆1,481Updated this week
 - PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆330Updated 3 years ago
 - ☆888Updated 2 months ago
 - Optax is a gradient processing and optimization library for JAX.☆2,065Updated this week
 - TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆991Updated this week
 - A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆883Updated this week
 - Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,001Updated 3 weeks ago
 - Orbax provides common checkpointing and persistence utilities for JAX users☆439Updated this week
 - Deep learning in Rust, with shape checked tensors and neural networks☆1,855Updated last year