m-wojnar / reinforced-lib

Related projects ⓘ

Alternatives and complementary repositories for reinforced-lib

• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆244Updated this week
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆164Updated 3 months ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆430Updated last week
• rbardenet / bml-course
  Public repo for course material on Bayesian machine learning at ENS Paris-Saclay and Univ Lille
  ☆74Updated 7 months ago
• harrygcoppock / ImperialMScAIUtils
  A repo containing useful things for the MSc AI course at Imperial
  ☆30Updated 11 months ago
• probabilists / zuko
  Normalizing flows in PyTorch
  ☆326Updated 3 weeks ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆153Updated this week
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆88Updated last year
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆102Updated 3 months ago
• srush / Autodiff-Puzzles
  ☆388Updated 3 weeks ago
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆89Updated 10 months ago
• google-deepmind / nanodo
  ☆197Updated 3 months ago
• jeremiecoullon / jax-tqdm
  Add a tqdm progress bar to your JAX scans and loops.
  ☆93Updated 2 weeks ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆521Updated this week
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆59Updated 3 weeks ago
• google-deepmind / distrax
  ☆535Updated last month
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆231Updated this week
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆102Updated 3 weeks ago
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆17Updated 5 months ago
• google / grain
  ☆296Updated this week
• eemlcommunity / PracticalSessions2022
  Hands-on tutorials at EEML2022 summer school
  ☆61Updated 2 years ago
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆34Updated 2 months ago
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆516Updated this week
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆361Updated last year
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆108Updated 6 months ago
• google-deepmind / mathematics_conjectures
  ☆367Updated 2 years ago
• Bogacz-Group / PredictiveCoding
  ☆16Updated 5 months ago
• jax-ml / oryx
  Oryx is a library for probabilistic programming and deep learning built on top of Jax.
  ☆219Updated last week
• ComputationalEvolutionaryMorphometry / hyperiax
  Tree traversals using JAX
  ☆9Updated 2 weeks ago
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆324Updated this week