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.
  ☆419Updated last month
• google / fiddle
  ☆348Updated 3 weeks ago
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆857Updated 2 weeks ago
• google / langfun
  OO for LLMs
  ☆779Updated last week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆927Updated this week
• Lightning-AI / litData
  Transform datasets at scale. Optimize datasets for fast AI model training.
  ☆485Updated this week
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆518Updated this week
• HMUNACHI / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆287Updated 9 months ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆382Updated last week
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,562Updated last month
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆586Updated last week
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆650Updated last year
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆890Updated last year
• mle-infrastructure / mle-monitor
  Lightweight Experiment & Resource Monitoring 📺
  ☆187Updated 2 years ago
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,422Updated last month
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆185Updated this week
• genomoncology / FuzzTypes
  Pydantic extension for annotating autocorrecting fields.
  ☆220Updated 11 months ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,784Updated last month
• 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,358Updated this week
• mit-ll-responsible-ai / hydra-zen
  Create powerful Hydra applications without the yaml files and boilerplate code.
  ☆385Updated 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…
  ☆499Updated 2 weeks ago
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆955Updated last month
• google / etils
  Collection of eclectic utils for python.
  ☆224Updated last week
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆177Updated last month
• facebookresearch / recipes
  Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…
  ☆318Updated last week
• google-deepmind / uncertain_ground_truth
  Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…
  ☆648Updated last year
• 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…
  ☆288Updated 6 months ago
• AI-Hypercomputer / RecML
  ☆179Updated this week
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆270Updated this week
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆331Updated last year