Alternatives and similar repositories for tx2

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

• SarthakYadav / audax
  A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.
  ☆69Updated 3 years ago
• google-deepmind / dm_aux
  ☆66Updated last year
• adlpr / iso4
  ISO 4 abbreviation of publication titles in Python
  ☆20Updated 2 years ago
• Roger-luo / pytorch-complex
  complex tensor plugin for pytorch (deprecated)
  ☆47Updated 6 years ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆26Updated 2 years ago
• google-deepmind / einshape
  ☆107Updated last year
• chrvt / denoising-normalizing-flow
  ☆20Updated 2 years ago
• sssohrab / rho_VAE
  rho_VAE: an autoregressive parametrization of the VAE encoder
  ☆16Updated 6 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• mbilos / stribor
  Library for normalizing flows and neural flows.
  ☆25Updated 3 years ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆32Updated 5 years ago
• xor2k / npy-append-array
  Create Numpy .npy files by appending on the growth axis
  ☆65Updated 4 months ago
• rsokl / phantom-tensors
  Tensor-like types – with variadic shapes – that support both static and runtime type checking, and convenient parsing
  ☆19Updated last month
• Lumi-supercomputer / LUMI-AI-Guide
  The LUMI AI Guide is designed to assist users in migrating their machine learning applications from smaller-scale computing environments …
  ☆60Updated last week
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆51Updated 4 years ago
• SkafteNicki / pl_crossvalidate
  ☆63Updated last year
• pnnl / DeepDataProfiler
  Python library for analyzing the internal structure of deep neural networks.
  ☆19Updated 2 months ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 11 months ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 5 years ago
• guillaumeBellec / multitask
  ☆24Updated last year
• dwromero / wavelet_networks
  Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs…
  ☆83Updated last year
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated 2 years ago
• keunwoochoi / tokenizer-vs-tokenizer
  ☆14Updated 2 years ago
• yukara-ikemiya / minimal-sqvae
  A minimal Pytorch Implementation of Stochastically Quantized Variational AutoEncoder (SQ-VAE) by Sony
  ☆33Updated 2 years ago
• SymJAX / SymJAX
  Documentation:
  ☆124Updated 2 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 9 months ago
• earthspecies / spectral_hyperresolution
  ☆23Updated 4 years ago
• google-research / soft-dtw-divergences
  An implementation of soft-DTW divergences.
  ☆135Updated 4 years ago
• francois-rozet / torchist
  NumPy-style histograms in PyTorch
  ☆55Updated last month
• namedtensor / notation
  ☆108Updated 2 years ago