Alternatives and similar repositories for mlsocket

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

• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• kazukiosawa / ngd_in_wide_nn
  simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…
  ☆14Updated 4 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆83Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 6 months ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago
• NERSC / slurm-ray-cluster
  ☆59Updated last year
• google-deepmind / tf2jax
  ☆115Updated this week
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆51Updated last week
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆112Updated 3 years ago
• RobertTLange / code-and-blog
  Some small scale experiments for my blog posts 📝
  ☆79Updated 3 years ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 5 years ago
• google-deepmind / jaxline
  ☆157Updated last year
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆10Updated 4 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆107Updated last year
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• RaulPL / awesome-gaussian-processes
  A curated list of resources for learning Gaussian Processes
  ☆40Updated 4 years ago
• aidanscannell / GPJax
  Minimal Gaussian process library in JAX with a simple (custom) approach to state management.
  ☆12Updated last year
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆38Updated 2 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆106Updated 3 weeks ago
• lukasheinrich / generative_models_examples
  playground for generative models
  ☆10Updated last year
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆29Updated 4 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆49Updated this week
• captain-pool / hydra-example
  ☆21Updated 3 years ago
• mazy1998 / Solving-Optimization-Problems-with-JAX
  Solving Optimization Problems with JAX, code and PDF
  ☆16Updated 5 years ago
• PredictiveIntelligenceLab / JAX-BO
  ☆30Updated 2 years ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated 2 years ago
• facebookresearch / how-to-autorl
  Plug-and-play hydra sweepers for the EA-based multifidelity method DEHB and several population-based training variations, all proven to e…
  ☆82Updated last year
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆34Updated last year