mit-ll-responsible-ai / hydra-zen-examples

Related projects: ⓘ

• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆10Updated 3 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆59Updated last year
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆27Updated 3 years ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆36Updated last year
• jax-ml / jax-ai-stack
  ☆28Updated this week
• saturncloud / dask-pytorch-ddp
  dask-pytorch-ddp is a Python package that makes it easy to train PyTorch models on dask clusters using distributed data parallel.
  ☆57Updated 3 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆31Updated 3 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆25Updated 2 years ago
• Lightning-AI / ecosystem-ci
  Automate issue discovery for your projects against Lightning nightly and releases.
  ☆45Updated last week
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated 8 months ago
• araffin / datasaurust
  Blazingly fast implementation of the Datasaurus paper. Same Stats, Different Graphs.
  ☆19Updated last year
• suneeta-mall / hydra_pydantic_config_management
  ☆16Updated 2 years ago
• conda-forge / jaxlib-feedstock
  A conda-smithy repository for jaxlib.
  ☆16Updated 3 weeks ago
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• Joshuaalbert / jaxnlds
  Inference on non-linear dynamical systems written in JAX
  ☆12Updated 4 years ago
• yuanqing-wang / galax
  Graph Learning with JAX
  ☆14Updated 2 years ago
• fastai / docments
  Document parameters using comments
  ☆10Updated 3 years ago
• lucidrains / viewformer-pytorch
  ☆10Updated this week
• probml / JSL
  Jax SSM Library
  ☆51Updated last year
• mhlr / awesome-jax
  List of awesome JAX resources
  ☆12Updated last year
• ThomasMiconi / Meta-Task-Generator
  Automatically generate simple meta-learning tasks from a very large space
  ☆15Updated last year
• Farama-Foundation / Jumpy
  On-the-fly conversions between Jax and NumPy tensors
  ☆46Updated last year
• mimoralea / gfootball-rllib-intro
  ☆17Updated 3 years ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 2 years ago
• microsoft / python-import-mocker
  Easily import a module and mock its dependencies in an isolated way.
  ☆13Updated 2 years ago
• natolambert / plotting-basics
  Collection of lines of code for basics of clean plots in Plotly and Matplotlib
  ☆13Updated 3 years ago
• bwasti / better_bindings
  Better bindings for Python
  ☆17Updated last year
• tchaton / pytorch2lightning
  ☆15Updated 3 years ago
• lucidrains / hopular-pytorch
  ☆21Updated this week