stackav-oss / dltypeLinks
Deep Learning Type Library
☆36Updated this week
Alternatives and similar repositories for dltype
Users that are interested in dltype are comparing it to the libraries listed below
Sorting:
- Fast dataset format and loader☆23Updated last month
- Easily run PyTorch on multiple GPUs & machines☆57Updated 3 weeks ago
- FID computation in Jax/Flax.☆29Updated last year
- Collection of lines of code for basics of clean plots in Plotly and Matplotlib☆13Updated 5 years ago
- Automatically take good care of your preemptible TPUs☆37Updated 2 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- Train vision models using JAX and 🤗 transformers☆100Updated last month
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- A simple, performant and scalable JAX-based world modeling codebase.☆129Updated 3 weeks ago
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 5 months ago
- Implementation of Reinforce for educational purposes.☆12Updated 2 years ago
- Lightweight ML Experiment Logging 📖☆80Updated last year
- ML/DL Math and Method notes☆66Updated 2 years ago
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆52Updated last week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆98Updated 6 months ago
- ☆22Updated 4 years ago
- Image augmentation library for Jax☆41Updated last year
- A natively parallel dataloader for Python, written in Rust. Serving data at GB/s speeds, while covering aspect ratio bucketing, crop and …☆126Updated last week
- WandB sweeps integration with Hydra sweeper☆50Updated 2 years ago
- Tools and Utils for Experiments (TUX)☆15Updated last year
- Test pytorch code with minimal computational overhead☆26Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Updated 2 years ago
- ☆31Updated 2 weeks ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Scalable Computation of Hessian Diagonals☆14Updated last year
- GPT implementation in Flax☆18Updated 4 years ago
- ☆123Updated 7 months ago
- A holistic evaluation library for multi-modal generative models using Weave☆27Updated last year
- Unofficial JAX implementations of deep learning research papers☆161Updated 3 years ago