BlackHC / neural_net_checklistView external linksLinks
☆215Aug 14, 2024Updated last year
Alternatives and similar repositories for neural_net_checklist
Users that are interested in neural_net_checklist are comparing it to the libraries listed below
Sorting:
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆187Jan 19, 2026Updated 3 weeks ago
- Automatic identification of regions in the latent space of a model that correspond to unique concepts, namely to concepts with a semantic…☆14Nov 22, 2023Updated 2 years ago
- ☆27May 3, 2024Updated last year
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Jun 11, 2023Updated 2 years ago
- ☆11Oct 15, 2025Updated 4 months ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆19Updated this week
- A dashboard for exploring timm learning rate schedulers☆19Nov 22, 2024Updated last year
- ☆56Nov 6, 2024Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆141Dec 20, 2023Updated 2 years ago
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Mar 21, 2023Updated 2 years ago
- Uncertainty quantification with PyTorch☆379Jan 27, 2026Updated 2 weeks ago
- ☆60Mar 8, 2022Updated 3 years ago
- 🧱 Modula software package☆322Aug 18, 2025Updated 5 months ago
- NeuMeta transforms neural networks by allowing a single model to adapt on the fly to different sizes, generating the right weights when n…☆44Nov 8, 2024Updated last year
- Tensors, for human consumption☆1,353Jan 22, 2026Updated 3 weeks ago
- Run PyTorch in JAX. 🤝☆312Oct 13, 2025Updated 4 months ago
- Quantification of Uncertainty with Adversarial Models☆29Jul 11, 2023Updated 2 years ago
- High-performance tokenized language data-loader for Python C++ extension☆14Jul 22, 2024Updated last year
- Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inferenc…☆11Dec 19, 2025Updated last month
- Minimal module for computing audio spectrograms☆15Feb 28, 2019Updated 6 years ago
- Code that accompanies the paper Guided Deep Kernel Learning☆10Jul 2, 2023Updated 2 years ago
- Sharpened Cosine Distance implementation in PyTorch☆10Feb 1, 2022Updated 4 years ago
- Focused on fast experimentation and simplicity☆80Dec 24, 2024Updated last year
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆53Apr 26, 2024Updated last year
- Efficient optimizers☆283Dec 20, 2025Updated last month
- utilities to facilitate working with codebases that don't ascribe to normal package management paradigms, e.g. ML research code that can …☆13Nov 26, 2022Updated 3 years ago
- ☆13Jul 12, 2024Updated last year
- PyTorch implementation of the paper Learning Multi-Level Representations for Hierarchical Music Structure Analysis presented at ISMIR 202…☆14Jan 2, 2023Updated 3 years ago
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…☆15Oct 16, 2023Updated 2 years ago
- ☆14Oct 17, 2023Updated 2 years ago
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- ☆11Dec 11, 2024Updated last year
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆29Jun 28, 2024Updated last year
- Minimal Implementation of a D3PM in pytorch☆286Apr 22, 2024Updated last year
- GPT for FACodec☆13Mar 25, 2024Updated last year
- ☆14Oct 18, 2023Updated 2 years ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- Algorithm to simulate resection surgery on brain MRI scans. Published in MICCAI 2020, extended in IJCARS.☆14Jan 22, 2026Updated 3 weeks ago