araffin / datasaurust

Related projects ⓘ

Alternatives and complementary repositories for datasaurust

• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 2 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• cgarciae / einop
  ☆58Updated 2 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• zach-lawless / gym-wordle
  Gym environment for playing Wordle with RL agents
  ☆38Updated 2 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆40Updated 2 years ago
• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 3 years ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆21Updated 3 weeks ago
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated 4 months ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆30Updated 11 months ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• ThomasMiconi / Meta-Task-Generator
  Automatically generate simple meta-learning tasks from a very large space
  ☆15Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆39Updated last week
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆20Updated 5 months ago
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆64Updated 4 months ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 3 weeks ago
• Michael-Beukman / RobocupGym
  Reinforcement Learning inside a 3D soccer simulation
  ☆25Updated 2 months ago
• Kajiyu / kanerva_machine
  The implementation of "The Kanerva Machine" with Pytorch and Pyro
  ☆12Updated 6 years ago
• yuanqing-wang / galax
  Graph Learning with JAX
  ☆14Updated 2 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆26Updated 2 years ago
• dasayan05 / iclr24_blog_code
  Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
  ☆13Updated 11 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• Rose-STL-Lab / AutoSTPP
  Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…
  ☆24Updated last month