mit-ll-responsible-ai / hydra-zen

Related projects ⓘ

Alternatives and complementary repositories for hydra-zen

• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• pytorch / hydra-torch
  Configuration classes enabling type-safe PyTorch configuration for Hydra apps
  ☆205Updated 2 years ago
• lebrice / SimpleParsing
  Simple, Elegant, Typed Argument Parsing with argparse
  ☆432Updated 3 weeks 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,219Updated this week
• omni-us / jsonargparse
  Implement minimal boilerplate CLIs derived from type hints and parse from command line, config files and environment variables
  ☆325Updated this week
• 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…
  ☆216Updated last week
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆470Updated last year
• eladrich / pyrallis
  Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure …
  ☆198Updated 11 months ago
• mle-infrastructure / mle-monitor
  Lightweight Experiment & Resource Monitoring 📺
  ☆187Updated last year
• xl0 / lovely-numpy
  NumPy arrays, ready for human consumption
  ☆63Updated 4 months ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 2 months ago
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆211Updated last year
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆285Updated 5 months ago
• merantix-momentum / squirrel-core
  A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way
  ☆282Updated this week
• google / grain
  ☆303Updated this week
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆328Updated last month
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆46Updated 9 months ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• fidelity / spock
  spock is a framework that helps manage complex parameter configurations during research and development of Python applications
  ☆124Updated last year
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆840Updated last week
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆113Updated last year
• j3soon / tbparse
  Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow
  ☆181Updated 3 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆102Updated 4 months ago
• aimhubio / aimlflow
  aim-mlflow integration
  ☆193Updated last year
• google-deepmind / tf2jax
  ☆105Updated 2 weeks ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• google / fiddle
  ☆328Updated this week
• google-deepmind / tensor_annotations
  Annotating tensor shapes using Python types
  ☆158Updated last year
• google / etils
  Collection of eclectic utils for python.
  ☆193Updated this week
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆103Updated this week