• Stars
    star
    61
  • Rank 494,050 (Top 10 %)
  • Language
    Python
  • License
    MIT License
  • Created over 1 year ago
  • Updated 3 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

A dead simple Python package for creating custom JAX pytree objects

More Repositories

1

pypeln

Concurrent data pipelines in Python >>>
Python
1,546
star
2

treex

A Pytree Module system for Deep Learning in JAX
Python
215
star
3

karma

An MVC framework for Unity3D
C#
167
star
4

phi

Functional Programming + Python - Pain
Python
132
star
5

tensorbuilder

TensorBuilder is a TensorFlow library enables you to easily create complex deep neural networks by leveraging the phi DSL to help define their structure.
Python
92
star
6

ciclo

A functional training loops library for JAX
Python
83
star
7

nnx

Neural Networks for JAX
Python
82
star
8

NDArray

A Multidimensional Array library for Swift
Swift
64
star
9

treeo

A small library for creating and manipulating custom JAX Pytree classes
Python
59
star
10

cybrain

Neural Networks in Cython, inspired by PyBrain.
HTML
58
star
11

einop

Python
56
star
12

jax_metrics

A metrics library for the JAX ecosystem
Python
35
star
13

Stream

A simple data processing library for Swift
Swift
28
star
14

jax-gravity

Jupyter Notebook
24
star
15

quantile-regression

Jupyter Notebook
19
star
16

dataget

A framework-agnostic datasets library for Machine Learning research and education.
Python
18
star
17

dyn_plot

Jupyter Notebook
13
star
18

resume

Machine Learning Engineer with background in Maths & Physics
CSS
9
star
19

simple-detr

Python
8
star
20

python_path

Python
7
star
21

stop_iter

Python
7
star
22

umvc

C#
6
star
23

TypeJS

Automatic Type-checking for JavaScript Functions!
JavaScript
5
star
24

jax-streams

Jupyter Notebook
5
star
25

simple-self-supervised

Python
4
star
26

mnist-estimator

Jupyter Notebook
4
star
27

refx

Python
4
star
28

algorithms

Python
3
star
29

object-detection

Jupyter Notebook
3
star
30

simple-ddpm

Jupyter Notebook
3
star
31

aoc-2023

Mojo
3
star
32

synthetic-casino

Shell
2
star
33

flax-tools

Python
2
star
34

jax-differentiable

Python
2
star
35

simple-mixture-models

Python
2
star
36

simple-attention

Python
2
star
37

NeuralJS

Neural Networks in JavaScript, inspired by PyBrain.
JavaScript
2
star
38

swift-test

Swift
1
star
39

tree_collections

Rust
1
star
40

tensorrl

Python
1
star
41

ml-meetup-1-intro-tensorflow

1
star
42

flax-xor-demo

Jupyter Notebook
1
star
43

simple-perceiver

Python
1
star
44

point-cloud-attention

Python
1
star
45

holberton

Python
1
star
46

tfinterface

Python
1
star
47

supervised-avanzado-german-traffic-signs

Jupyter Notebook
1
star
48

udacity-simulator-dataset

Python
1
star
49

feature-engineering-blog

Jupyter Notebook
1
star
50

nfl

Jupyter Notebook
1
star
51

Atoms

molecules
C#
1
star
52

test_tf2

Jupyter Notebook
1
star
53

live-q-table

Jupyter Notebook
1
star
54

arista-server

JavaScript
1
star
55

domain-randomization

Python
1
star
56

media

1
star