stackav-oss / dltypeLinks
Deep Learning Type Library
☆33Updated 2 weeks ago
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 10 months ago
- Automatically take good care of your preemptible TPUs☆37Updated 2 years ago
- FID computation in Jax/Flax.☆29Updated last year
- Train vision models using JAX and 🤗 transformers☆100Updated 3 weeks ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- PyTorch centric eager mode debugger☆48Updated 11 months ago
- Lightweight ML Experiment Logging 📖☆81Updated last year
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- WandB sweeps integration with Hydra sweeper☆50Updated last year
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆52Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 10 months ago
- Easily run PyTorch on multiple GPUs & machines☆53Updated last week
- TorchFix - a linter for PyTorch-using code with autofix support☆151Updated 3 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 5 months ago
- ☆31Updated last week
- A simple, performant and scalable JAX-based world modeling codebase☆111Updated 3 weeks ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated last year
- Gym environment for playing Wordle with RL agents☆43Updated 3 years ago
- Code for minimum-entropy coupling.☆32Updated last year
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 4 years ago
- Utilities for Training Very Large Models☆58Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆116Updated 3 years ago
- ☆21Updated 8 months ago
- ☆18Updated last year
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- ☆119Updated 5 months ago
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆119Updated last week
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Updated last year
- Building blocks for productive research☆63Updated 3 months ago