ivy-llc / mech

Related projects ⓘ

Alternatives and complementary repositories for mech

• huggingface / hf_benchmarks
  A starter kit for evaluating benchmarks on the 🤗 Hub
  ☆13Updated 10 months ago
• facebookresearch / flashy
  Framework for writing deep learning training loops. Lightweight, and retaining full freedom to design as you see fits. It handles checkpo…
  ☆103Updated 7 months ago
• google / etils
  Collection of eclectic utils for python.
  ☆193Updated 2 weeks ago
• chainyo / tensorshare
  🤝 Trade any tensors over the network
  ☆30Updated last year
• Oxid15 / cascade
  Lightweight and modular MLOps library targeted at small teams or individuals
  ☆22Updated this week
• ClashLuke / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆39Updated last year
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆112Updated last year
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆185Updated 2 years ago
• HomebrewNLP / HomebrewNLP
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 2 years ago
• fchollet / namex
  Clean up the public namespace of your package!
  ☆52Updated 6 months ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆48Updated last month
• merantix-momentum / squirrel-datasets-core
  Squirrel dataset hub
  ☆42Updated last year
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆24Updated 3 months ago
• LukasHedegaard / ride
  Training wheels, side rails, and helicopter parent for your Deep Learning projects in PyTorch
  ☆25Updated last year
• srush / g9py
  ☆18Updated last year
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆66Updated last year
• msaroufim / torchprep
  ☆11Updated 2 years ago
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆52Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆42Updated 5 months ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆46Updated 9 months ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆95Updated 3 weeks ago
• sholtodouglas / scalingExperiments
  ☆56Updated 2 years ago
• huggingface / hffs
  **ARCHIVED** Filesystem interface to 🤗 Hub
  ☆56Updated last year
• Borda / pyDeprecate
  Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.
  ☆50Updated 3 weeks ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated last week
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆37Updated 2 years ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated last year