Alternatives and similar repositories for bagz

Users that are interested in bagz are comparing it to the libraries listed below

• google / array_record
  ☆129Updated this week
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆126Updated 3 months ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆352Updated this week
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆101Updated last month
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆456Updated 2 weeks ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆202Updated this week
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆48Updated last year
• openai / chz
  ☆205Updated 3 weeks ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆46Updated 2 years ago
• AI-Hypercomputer / maxdiffusion
  ☆290Updated this week
• google / praxis
  ☆190Updated last week
• google / torchax
  torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…
  ☆148Updated 2 weeks ago
• google / aqt
  ☆340Updated last week
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆193Updated last year
• microsoft / dion
  Dion optimizer algorithm
  ☆404Updated this week
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆132Updated 5 months ago
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆97Updated 4 months ago
• google / qwix
  a Jax quantization library
  ☆74Updated last week
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆121Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆436Updated last week
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆368Updated this week
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆332Updated 2 years ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆552Updated last week
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last month
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆169Updated 4 months ago
• riverstone496 / awesome-second-order-optimization
  ☆27Updated 2 months ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆309Updated 2 months ago
• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆453Updated 8 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆320Updated this week