google / mediapy

Related projects ⓘ

Alternatives and complementary repositories for mediapy

• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆389Updated last week
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• assafshocher / ResizeRight
  The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).
  ☆548Updated last year
• google / fiddle
  ☆328Updated this week
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆321Updated this week
• airalcorn2 / pytorch-nerf
  Minimal PyTorch implementations of NeRF and pixelNeRF.
  ☆233Updated last year
• ankurhanda / nerf2D
  Adding positional encoding to the input preserves sharp edges in the image
  ☆206Updated 2 years ago
• brohrer / sharpened-cosine-similarity
  An alternative to convolution in neural networks
  ☆251Updated 7 months ago
• brentyi / tyro
  CLI interfaces & config objects, from types
  ☆526Updated this week
• naver / roma
  RoMa: A lightweight library to deal with 3D rotations in PyTorch.
  ☆515Updated 3 weeks ago
• 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
• google-deepmind / chex
  ☆788Updated this week
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆896Updated this week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆200Updated last year
• iejMac / video2numpy
  Optimized library for large-scale extraction of frames and audio from video.
  ☆203Updated last year
• aharley / pips
  Particle Video Revisited
  ☆569Updated last year
• unixpickle / sk2torch
  Convert scikit-learn models to PyTorch modules
  ☆157Updated 6 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…
  ☆294Updated this week
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆473Updated 2 years ago
• facebookresearch / moolib
  A library for distributed ML training with PyTorch
  ☆366Updated last year
• google-deepmind / distrax
  ☆536Updated 2 months ago
• apple / ml-gsn
  ☆297Updated 3 years ago
• aliutkus / torchsearchsorted
  Pytorch Custom CUDA kernel for searchsorted
  ☆135Updated last year
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆470Updated last year
• google-research / visu3d
  3d without friction (Torch, TF, Jax, Numpy)
  ☆147Updated this week
• krrish94 / nerf-pytorch
  A PyTorch re-implementation of Neural Radiance Fields
  ☆887Updated 3 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,219Updated this week
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆238Updated last year