facebookresearch / moolib

Related projects: ⓘ

• facebookresearch / rlmeta
  RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.
  ☆284Updated last year
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆379Updated 2 months ago
• facebookresearch / salina
  a Lightweight library for sequential learning agents, including reinforcement learning
  ☆426Updated last year
• google-deepmind / launchpad
  ☆309Updated last year
• facebookresearch / rlstructures
  RLStructures is a library to facilitate the implementation of new reinforcement learning algorithms. It includes a library, a tutorial, a…
  ☆259Updated last year
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆814Updated last month
• google / fiddle
  ☆320Updated this week
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆211Updated 10 months ago
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆236Updated last year
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• chandar-lab / RLHive
  ☆100Updated 7 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…
  ☆286Updated this week
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• google-deepmind / jaxline
  ☆149Updated 9 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆240Updated last week
• google-deepmind / chex
  ☆759Updated 2 months ago
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆474Updated 2 years ago
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆252Updated last year
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆468Updated last year
• google-deepmind / distrax
  ☆529Updated 2 months ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,389Updated this week
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆546Updated 3 years ago
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆318Updated 3 weeks ago
• facebookresearch / minihack
  MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research
  ☆471Updated last month
• salesforce / warp-drive
  Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)
  ☆457Updated last month
• huggingface / simulate
  🎢 Creating and sharing simulation environments for embodied and synthetic data research
  ☆187Updated 11 months ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆321Updated 2 weeks ago
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆879Updated last month
• google / flaxformer
  ☆322Updated 5 months ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆215Updated last year