Alternatives and similar repositories for neural_networks_and_computation

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

• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆58Updated last year
• aradha / recursive_feature_machines
  ☆55Updated 2 months ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆49Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆177Updated 2 weeks ago
• wgrathwohl / GWG_release
  Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"
  ☆55Updated last year
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆100Updated last year
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆87Updated 3 months ago
• lindermanlab / elk
  Scalable and Stable Parallelization of Nonlinear RNNS
  ☆16Updated 4 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• AllanYangZhou / universal_neural_functional
  ☆51Updated last year
• google-research / discs
  DISCS: The code base for the Benchmark for Discrete Sampling
  ☆56Updated 10 months ago
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆72Updated 3 months ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆78Updated 2 years ago
• modula-systems / modula
  🧱 Modula software package
  ☆200Updated 2 months ago
• requeima / llm_processes
  ☆60Updated 2 months ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆66Updated last month
• mila-iqia / milatools
  Tools to connect to and interact with the Mila cluster
  ☆72Updated last week
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆47Updated last year
• normal-computing / thermox
  Exact OU processes with JAX
  ☆48Updated 3 months ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 6 months ago
• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆40Updated 2 months ago
• google-deepmind / nanodo
  ☆270Updated 11 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆87Updated last month
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆58Updated 2 years ago
• necludov / jam
  Implementation of Action Matching
  ☆44Updated 2 years ago
• google-deepmind / annealed_flow_transport
  ☆52Updated 2 years ago
• jamesvuc / jax-bayes
  Bayesian inference with Python and Jax.
  ☆32Updated 2 years ago
• lgresele / independent-mechanism-analysis
  Code for the paper: "Independent mechanism analysis, a new concept?"
  ☆24Updated last year