Alternatives and similar repositories for wandb-workspaces:

Users that are interested in wandb-workspaces are comparing it to the libraries listed below

• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆47Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 2 months ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆25Updated last month
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 6 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆90Updated last year
• danijar / elements
  Building blocks for productive research
  ☆51Updated 2 months ago
• Artur-Galstyan / statedict2pytree
  ☆42Updated last week
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 8 months ago
• JoeyTeng / jaxrenderer
  Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…
  ☆71Updated last year
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆57Updated 10 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated last year
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• danijar / portal
  Fast and reliable distributed systems in Python
  ☆25Updated last month
• cgarciae / einop
  ☆60Updated 3 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆62Updated 8 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆134Updated 5 months ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated this week
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆175Updated last week
• google-deepmind / einshape
  ☆101Updated 9 months ago
• google-deepmind / tf2jax
  ☆113Updated this week
• wookayin / expt
  Experiment. Plot. Tabulate.
  ☆69Updated 7 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆232Updated last month
• jax-ml / jax-llm-examples
  ☆87Updated 2 weeks ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 2 months ago
• google-deepmind / hierarchical_perceiver
  ☆26Updated 8 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆26Updated 9 months ago
• geyang / jaynes-starter-kit
  a starter-kit for jaynes, the cloud-agnostic launch library
  ☆17Updated 7 months ago
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆30Updated 2 years ago
• danijar / granular
  Fast dataset format and loader
  ☆22Updated 2 months ago