Alternatives and similar repositories for tx2

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

• ORNL / curifactory
  An experiment workflow and organization tool.
  ☆18Updated 8 months ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated 2 years ago
• SkafteNicki / pl_crossvalidate
  ☆61Updated last year
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated last year
• SiggiGue / hdfdict
  Helps h5py to dump and load dictionaries.
  ☆22Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 5 years ago
• dionhaefner / bayesian-histograms
  Bayesian histograms for estimation of binary rare event rates, with fully automated bin pruning
  ☆28Updated 3 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• androstj / tensor_rnn
  An implementation of various tensor-based decomposition for NN & RNN parameters
  ☆18Updated 7 years ago
• TutteInstitute / vectorizers
  Vectorizers for a range of different data types
  ☆102Updated 4 months ago
• mbilos / stribor
  Library for normalizing flows and neural flows.
  ☆24Updated 3 years ago
• mcbal / deep-implicit-attention
  Implementation of deep implicit attention in PyTorch
  ☆65Updated 3 years ago
• pierreablin / qndiag
  Quasi-Newton algorithm for joint-diagonalization
  ☆32Updated 2 years ago
• craffel / jax-tutorial
  A tutorial on JAX (https://github.com/google/jax/)
  ☆46Updated 6 years ago
• aspuru-guzik-group / funsies
  funsies is a lightweight workflow engine 🔧
  ☆41Updated 3 years ago
• stfc-sciml / sciml-bench
  SciML Benchmarking Suite for AI for Science
  ☆41Updated 11 months ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• ElsevierSoftwareX / SOFTX_2019_95
  Machine learning algorithms for many-body quantum systems. To cite this software publication: https://www.sciencedirect.com/science/artic…
  ☆10Updated 5 years ago
• suriyadeepan / pm-pyro
  PyMC3-like Interface for Pyro
  ☆37Updated 5 years ago
• NERSC / slurm-ray-cluster
  ☆57Updated last year
• Quansight-Labs / unumpy
  A backend-dispatchable version of NumPy.
  ☆19Updated 4 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆43Updated 4 years ago
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated last year
• SymJAX / SymJAX
  Documentation:
  ☆120Updated 2 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• NeuralEnsemble / lazyarray
  lazyarray is a Python package that provides a lazily-evaluated numerical array class, larray, based on and compatible with NumPy arrays.
  ☆21Updated 9 months ago
• mcbal / afem
  Implementation of approximate free-energy minimization in PyTorch
  ☆19Updated 3 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 8 months ago
• google-research / spherical-cnn
  ☆118Updated 6 months ago
• yuanqing-wang / galax
  Graph Learning with JAX
  ☆14Updated 2 years ago