loeweX / Custom-ConvLayers-PytorchLinks
A reimplementation of 2D Convolutional and Transposed Convolutional Layers in PyTorch, designed for easy modifications and analysis. Includes comprehensive explanations and testing.
☆21Updated last year
Alternatives and similar repositories for Custom-ConvLayers-Pytorch
Users that are interested in Custom-ConvLayers-Pytorch are comparing it to the libraries listed below
Sorting:
- Sparse and discrete interpretability tool for neural networks☆63Updated last year
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆59Updated last year
- ☆69Updated 11 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 7 months ago
- ☆56Updated 8 months ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆71Updated last month
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Updated last year
- ☆62Updated 9 months ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- ☆53Updated 10 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 6 months ago
- PyTorch implementation for MRL☆19Updated last year
- ☆150Updated 11 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 2 months ago
- Official Repository of Pretraining Without Attention (BiGS), BiGS is the first model to achieve BERT-level transfer learning on the GLUE …☆114Updated last year
- Functional local implementations of main model parallelism approaches☆96Updated 2 years ago
- Code for minimum-entropy coupling.☆32Updated last year
- Automatic identification of regions in the latent space of a model that correspond to unique concepts, namely to concepts with a semantic…☆14Updated last year
- nanoGPT-like codebase for LLM training☆102Updated 2 months ago
- ☆82Updated last year
- ☆21Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆85Updated last year
- Implementation of GateLoop Transformer in Pytorch and Jax☆89Updated last year
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- ☆26Updated 2 years ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆130Updated 2 years ago
- Quantification of Uncertainty with Adversarial Models☆30Updated 2 years ago