google / pyglove

Related projects ⓘ

Alternatives and complementary repositories for pyglove

• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆408Updated this week
• Lightning-AI / lightning-thunder
  Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors a…
  ☆1,187Updated this week
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆332Updated last week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆832Updated 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…
  ☆456Updated last week
• modularml / max
  A collection of sample programs, notebooks, and tools which highlight the power of the MAX Platform
  ☆364Updated this week
• apple / ml-sigma-reparam
  ☆292Updated 4 months ago
• dorjeduck / llm.mojo
  port of Andrjey Karpathy's llm.c to Mojo
  ☆321Updated 3 weeks ago
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆479Updated 2 weeks ago
• google / fiddle
  ☆328Updated this week
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆330Updated last week
• google-ai-edge / model-explorer
  A modern model graph visualizer and debugger
  ☆1,045Updated this week
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,475Updated this week
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆865Updated 6 months ago
• google-deepmind / recurrentgemma
  Open weights language model from Google DeepMind, based on Griffin.
  ☆606Updated 4 months ago
• endia-ai / Endia
  Build and train Neural Networks in Mojo
  ☆207Updated last month
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆274Updated 2 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆208Updated this week
• microsoft / Trace
  End-to-end Generative Optimization for AI Agents
  ☆322Updated this week
• mle-infrastructure / mle-monitor
  Lightweight Experiment & Resource Monitoring 📺
  ☆187Updated last year
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆817Updated 3 months ago
• ml-explore / mlx-data
  Efficient framework-agnostic data loading
  ☆362Updated 2 months ago
• NVIDIA-Merlin / dataloader
  The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
  ☆407Updated 6 months ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,010Updated 6 months ago
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆649Updated this week
• TilliFe / Infermo
  Tensors and dynamic Neural Networks in Mojo
  ☆124Updated 3 months ago
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆648Updated 7 months ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• Lightning-AI / litdata
  Transform datasets at scale. Optimize datasets for fast AI model training.
  ☆362Updated this week
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆516Updated this week