Alternatives and similar repositories for pyglove

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

• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆423Updated 2 months ago
• google / fiddle
  ☆350Updated last week
• google / langfun
  OO for LLMs
  ☆804Updated this week
• google / paxml
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆512Updated 2 weeks ago
• zeno-ml / zeno
  AI Data Management & Evaluation Platform
  ☆215Updated last year
• genomoncology / FuzzTypes
  Pydantic extension for annotating autocorrecting fields.
  ☆221Updated last year
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,454Updated 2 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆520Updated last week
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆892Updated last year
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,795Updated last week
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,572Updated this week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆931Updated last week
• nabla-ml / nabla
  Composable Function Transformations in Python with Mojo/MAX acceleration
  ☆254Updated this week
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆332Updated 2 years ago
• HMUNACHI / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆288Updated 10 months ago
• Lightning-AI / lightning-thunder
  Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …
  ☆1,369Updated this week
• dorjeduck / llm.mojo
  port of Andrjey Karpathy's llm.c to Mojo
  ☆352Updated 6 months ago
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆599Updated this week
• google-deepmind / onetwo
  ☆227Updated 4 months ago
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆859Updated last month
• google-deepmind / recurrentgemma
  Open weights language model from Google DeepMind, based on Griffin.
  ☆643Updated 3 weeks ago
• Slicer / light-the-torch
  Install PyTorch distributions with computation backend auto-detection
  ☆253Updated 2 months ago
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆475Updated 2 years ago
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆265Updated last year
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆384Updated this week
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆960Updated 2 months ago
• wandb / weave
  Weave is a toolkit for developing AI-powered applications, built by Weights & Biases.
  ☆914Updated this week
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆418Updated 3 months ago
• google / etils
  Collection of eclectic utils for python.
  ☆229Updated last week
• google-deepmind / nanodo
  ☆270Updated 11 months ago