Software Engineer at Google DeepMind working on JAX/Flax
pypeln. Concurrent data pipelines in Python >>>
1.6ktreex. A Pytree Module system for Deep Learning in JAX
212karma. An MVC framework for Unity3D
168phi. Functional Programming + Python - Pain
136ciclo. A functional training loops library for JAX
88tensorbuilder. TensorBuilder is a TensorFlow library enables you to easily create complex deep neural networks by leveraging the phi DSL to help define their structure.
88nnx. Neural Networks for JAX
84NDArray. A Multidimensional Array library for Swift
66simple-pytree. A dead simple Python package for creating custom JAX pytree objects
66einop. Python
61cybrain. Neural Networks in Cython, inspired by PyBrain.
58treeo. A small library for creating and manipulating custom JAX Pytree classes
56jax_metrics. A metrics library for the JAX ecosystem
41nanoGPT-jax. The simplest, fastest repository for training/finetuning medium-sized GPTs.
38Stream. A simple data processing library for Swift
28simple_flow_matching. Jupyter Notebook
24jax-gravity. Jupyter Notebook
24quantile-regression. Jupyter Notebook
21dataget. A framework-agnostic datasets library for Machine Learning research and education.
18dyn_plot. Jupyter Notebook
13simple-detr. Python
10resume. Machine Learning Engineer with background in Maths & Physics
9stop_iter. Python
8python_path. Python
7umvc. C#
6TypeJS. Automatic Type-checking for JavaScript Functions!
5jax-streams. Jupyter Notebook
5simple-self-supervised. Python
4refx. Python
4mnist-estimator. Jupyter Notebook
4simple-ddpm. Jupyter Notebook
3aoc-2023. Mojo
3algorithms. Python
3object-detection. Jupyter Notebook
3minGPT-jax. A minimal JAX re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
3synthetic-casino. Shell
2keras. Deep Learning for humans
2flax-tools. Python
2simple-mixture-models. Python
2Zenject. Dependency Injection Framework for Unity3D
2simple-attention. Python
2jax-differentiable. Python
2elegy. Elegy is a Neural Networks framework based on Jax and inspired by Keras.
1udacity-simulator-dataset. Python
1supervised-avanzado-german-traffic-signs. Jupyter Notebook
1media.
1tfinterface. Python
1ml-meetup-1-intro-tensorflow.
1tree_collections. Rust
1NDArray-clibs. System modules used internally by Nifty
1point-cloud-attention. Python
1