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.
  ☆68Updated 3 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆44Updated 4 years ago
• google-deepmind / einshape
  ☆105Updated last year
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• google-deepmind / dm_aux
  ☆66Updated last year
• ai4iacts / hexagdly
  Process hexagonally sampled data with PyTorch
  ☆94Updated 4 years ago
• pierreablin / qndiag
  Quasi-Newton algorithm for joint-diagonalization
  ☆32Updated 2 years ago
• TutteInstitute / vectorizers
  Vectorizers for a range of different data types
  ☆102Updated 6 months ago
• pnnl / DeepDataProfiler
  Python library for analyzing the internal structure of deep neural networks.
  ☆19Updated this week
• cgarciae / einop
  ☆60Updated 3 years ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 5 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 / synjax
  ☆247Updated 2 months ago
• gradhep / neos
  Upstream optimisation for downstream inference
  ☆70Updated 3 weeks ago
• chrvt / denoising-normalizing-flow
  ☆20Updated 2 years ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 9 months ago
• arogozhnikov / adamw_bfloat16
  AdamW optimizer for bfloat16 models in pytorch 🔥.
  ☆36Updated last year
• wil-j-wil / gp-time-freq
  Gaussian process time-frequency analysis toy examples
  ☆17Updated 6 years ago
• ag1988 / dlr
  The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…
  ☆22Updated 2 years ago
• SymJAX / SymJAX
  Documentation:
  ☆121Updated 2 years ago
• 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 …
  ☆53Updated 2 months ago
• mbilos / stribor
  Library for normalizing flows and neural flows.
  ☆25Updated 3 years ago
• sissa-data-science / DADApy
  Distance-based Analysis of DAta-manifolds in python
  ☆132Updated 2 months ago
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated last year
• mcbal / deep-implicit-attention
  Implementation of deep implicit attention in PyTorch
  ☆65Updated 4 years ago
• earthspecies / spectral_hyperresolution
  ☆23Updated 4 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 7 months ago
• shawwn / sparkvis
  Visualize tensors in a plain Python REPL using Sparklines
  ☆45Updated 4 years ago
• cifkao / jax-spectral
  Short-time Fourier transform (STFT) for JAX
  ☆14Updated 3 years ago