xl0 / lovely-numpy

Related projects ⓘ

Alternatives and complementary repositories for lovely-numpy

• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆174Updated 2 years ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• mit-ll-responsible-ai / hydra-zen
  Create powerful Hydra applications without the yaml files and boilerplate code.
  ☆338Updated this week
• pmeier / light-the-torch
  Install PyTorch distributions with computation backend auto-detection
  ☆220Updated last year
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆112Updated last year
• twsl / whos-there
  The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends
  ☆76Updated this week
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆284Updated 4 months ago
• mle-infrastructure / mle-monitor
  Lightweight Experiment & Resource Monitoring 📺
  ☆187Updated last year
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆151Updated 2 years ago
• jacobgil / confidenceinterval
  The long missing library for python confidence intervals
  ☆132Updated 5 months ago
• pytorch / hydra-torch
  Configuration classes enabling type-safe PyTorch configuration for Hydra apps
  ☆204Updated 2 years ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆104Updated last year
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆85Updated 8 months ago
• fchollet / namex
  Clean up the public namespace of your package!
  ☆52Updated 6 months ago
• jax-ml / oryx
  Oryx is a library for probabilistic programming and deep learning built on top of Jax.
  ☆219Updated last week
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆39Updated last year
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆102Updated 3 months ago
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆321Updated 6 months ago
• unixpickle / sk2torch
  Convert scikit-learn models to PyTorch modules
  ☆157Updated 5 months ago
• 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…
  ☆215Updated last week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆199Updated last year
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆326Updated 3 weeks ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆80Updated this week
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆88Updated last year
• google-deepmind / einshape
  ☆98Updated 4 months ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆388Updated this week
• LaihoE / did-it-spill
  Check if you have training samples in your test set
  ☆64Updated 2 years ago
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆211Updated last year