shawwn / jaxnotes
☆20Updated 3 years ago
Alternatives and similar repositories for jaxnotes:
Users that are interested in jaxnotes are comparing it to the libraries listed below
- ☆39Updated 2 years ago
- High-performance tokenized language data-loader for Python C++ extension☆12Updated 6 months ago
- ☆18Updated 9 months ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- Write your code as tree-like expressions, then transform it☆21Updated last year
- ☆59Updated 2 years ago
- One stop shop for all things carp☆59Updated 2 years ago
- List of awesome JAX resources☆13Updated 2 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated 8 months ago
- Lazy, a tool for running things in idle time☆48Updated 3 years ago
- A JAX nn library☆21Updated 11 months ago
- Test prompts for GPT-J-6B and the resulting AI-generated texts☆53Updated 3 years ago
- You should use PySR to find scaling laws. Here's an example.☆33Updated last year
- a writeup on some experiments on a sequence model for chess games☆28Updated 3 years ago
- Graph Learning with JAX☆14Updated 2 years ago
- Python Research Framework☆106Updated 2 years ago
- Implementation of a holodeck, written in Pytorch☆17Updated last year
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- ☆19Updated 4 years ago
- ☆21Updated last year
- A stateful pytree library for training neural networks.☆21Updated 2 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- A virtual dog-sitter that tracks, classifies, and responds to dog audio.☆32Updated 2 years ago
- ☆33Updated 4 years ago
- A programming language for formal/informal computation.☆41Updated this week
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- Geometric Algebra package for JAX☆49Updated 3 years ago
- ☆108Updated 2 years ago
- RWKV model implementation☆37Updated last year
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago