Alternatives and similar repositories for pyglove:

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

• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆501Updated this week
• google / langfun
  OO for LLMs
  ☆665Updated this week
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆398Updated last week
• google / fiddle
  ☆345Updated this week
• google-deepmind / recurrentgemma
  Open weights language model from Google DeepMind, based on Griffin.
  ☆627Updated last month
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,348Updated this week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆898Updated this week
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆261Updated last year
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆884Updated 11 months ago
• google / paxml
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆483Updated last week
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• chalk-diagrams / chalk
  A declarative drawing API in Python
  ☆294Updated 6 months ago
• Lightning-AI / litdata
  Transform datasets at scale. Optimize datasets for fast AI model training.
  ☆430Updated this week
• google / etils
  Collection of eclectic utils for python.
  ☆210Updated this week
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆557Updated this week
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,538Updated this week
• google-research / python-graphs
  A static analysis library for computing graph representations of Python programs suitable for use with graph neural networks.
  ☆332Updated last year
• zeno-ml / zeno
  AI Data Management & Evaluation Platform
  ☆215Updated last year
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆282Updated 6 months ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,746Updated 3 months ago
• facebookresearch / recipes
  Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…
  ☆310Updated this week
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆370Updated this week
• mle-infrastructure / mle-monitor
  Lightweight Experiment & Resource Monitoring 📺
  ☆187Updated 2 years ago
• pytorch / torchft
  PyTorch per step fault tolerance (actively under development)
  ☆267Updated this week
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆361Updated last month
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆172Updated this week
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆230Updated this week
• HazyResearch / meerkat
  Creative interactive views of any dataset.
  ☆837Updated 2 months ago
• pytorch / torcheval
  A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…
  ☆229Updated 2 months ago
• FlorianDietz / comgra
  A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…
  ☆282Updated 3 months ago