araffin / datasaurust

Related projects: ⓘ

• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆31Updated 3 years ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆60Updated 2 months ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 2 years ago
• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 3 years ago
• cgarciae / einop
  ☆56Updated 2 years ago
• srush / Tensor-Puzzles-Penzai
  ☆17Updated 4 months ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 2 years ago
• ThomasMiconi / Meta-Task-Generator
  Automatically generate simple meta-learning tasks from a very large space
  ☆15Updated last year
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆25Updated 2 years ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 2 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆34Updated last month
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• Joshuaalbert / jaxnlds
  Inference on non-linear dynamical systems written in JAX
  ☆12Updated 4 years ago
• maxencefaldor / Leniabreeder
  Repository for "Toward Artificial Open-Ended Evolution within Lenia using Quality-Diversity" (ALIFE 2024).
  ☆13Updated 2 months ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆59Updated last year
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆30Updated 9 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆19Updated 3 months ago
• glouppe / ssi2023
  ☆20Updated 9 months ago
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆25Updated 2 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆44Updated last year
• ElisevanderPol / PRAE
  Plannable Approximations to MDP Homomorphisms: Equivariance under Actions
  ☆26Updated 4 years ago
• geyang / plan2vec
  Public Release of Plan2vec Implementation in pyTorch
  ☆56Updated last year
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 3 years ago
• vicariousinc / PGMax
  Loopy belief propagation for factor graphs on discrete variables, in JAX!
  ☆62Updated 2 months ago
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated last month
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago