Alternatives and similar repositories for tx2

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

• pnnl / DeepDataProfiler
  Python library for analyzing the internal structure of deep neural networks.
  ☆19Updated 2 weeks ago
• SarthakYadav / audax
  A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.
  ☆70Updated 3 years ago
• google-deepmind / dm_aux
  ☆67Updated last year
• pierreablin / qndiag
  Quasi-Newton algorithm for joint-diagonalization
  ☆34Updated 3 years ago
• namedtensor / notation
  ☆108Updated 3 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 …
  ☆64Updated last week
• SkafteNicki / pl_crossvalidate
  ☆64Updated 2 years ago
• Roger-luo / pytorch-complex
  complex tensor plugin for pytorch (deprecated)
  ☆47Updated 6 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆47Updated 5 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 4 years ago
• adlpr / iso4
  ISO 4 abbreviation of publication titles in Python
  ☆20Updated 2 years ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆32Updated 5 years ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆26Updated 2 years ago
• wil-j-wil / gp-time-freq
  Gaussian process time-frequency analysis toy examples
  ☆17Updated 6 years ago
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated 2 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆55Updated 4 years ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆15Updated last year
• google-deepmind / einshape
  ☆109Updated last year
• mbilos / stribor
  Library for normalizing flows and neural flows.
  ☆26Updated 3 years ago
• cifkao / jax-spectral
  Short-time Fourier transform (STFT) for JAX
  ☆15Updated 4 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• google-research / spherical-cnn
  ☆130Updated this week
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆69Updated 4 years ago
• suriyadeepan / pm-pyro
  PyMC3-like Interface for Pyro
  ☆38Updated 6 years ago
• MilesCranmer / python_citations
  Bibtex for various Python science and machine learning software
  ☆33Updated 3 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 3 years ago
• ml-jku / quam
  Quantification of Uncertainty with Adversarial Models
  ☆29Updated 2 years ago
• jesmith14 / REAL-ML
  The Recognizing, Exploring, and Articulating Limitations in Machine Learning research tool (REAL ML) is a set of guided activities to hel…
  ☆52Updated 3 years ago
• earthspecies / spectral_hyperresolution
  ☆23Updated 4 years ago
• bdusell / semiring-einsum
  Generic PyTorch implementation of einsum that supports different semirings
  ☆51Updated 2 months ago