Lucasc-99 / NoTorch
A from-scratch neural network and transformers library, with speeds rivaling PyTorch
☆10Updated last year
Alternatives and similar repositories for NoTorch:
Users that are interested in NoTorch are comparing it to the libraries listed below
- Repository of machine learning benchmarks☆34Updated this week
- an environment based on XLA for deep learning compiler optimization research.☆23Updated last year
- Minimal deep learning library written from scratch in Python, using NumPy/CuPy.☆120Updated 2 years ago
- ☆211Updated 7 months ago
- OpTree: Optimized PyTree Utilities☆166Updated this week
- Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.☆24Updated 4 months ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆90Updated 2 months ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- seqax = sequence modeling + JAX☆143Updated 7 months ago
- Minimal but scalable implementation of large language models in JAX☆32Updated 3 months ago
- Hierarchical Associative Memory User Experience☆98Updated last year
- A functional training loops library for JAX☆86Updated last year
- Cost aware hyperparameter tuning algorithm☆143Updated 7 months ago
- Code associated to papers on superposition (in ML interpretability)☆27Updated 2 years ago
- A simple library for scaling up JAX programs☆129Updated 3 months ago
- JAX Synergistic Memory Inspector☆168Updated 7 months ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆56Updated 2 years ago
- Solve puzzles. Learn CUDA.☆62Updated last year
- ☆146Updated 2 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆53Updated 3 months ago
- A lightweight deep learning framework☆32Updated 3 years ago
- Named Tensors for Legible Deep Learning in JAX☆161Updated last week
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated 8 months ago
- Reimplementation of Geoffrey Hinton's Forward-Forward Algorithm☆142Updated last year
- 🦠 AD in less than 20 lines☆54Updated 3 years ago
- Procgen2: A community maintained fork of procgen☆11Updated 2 years ago
- Custom triton kernels for training Karpathy's nanoGPT.☆16Updated 4 months ago
- Module 0 - Fundamentals☆101Updated 5 months ago
- Parallel data preprocessing for NLP and ML.☆34Updated 3 months ago
- Various reinforcement learning algorithms written in Jax + Flax☆23Updated last year